假设有两个张量列表:
r1 = K.variable(1)
r2 = K.variable(2)
v1 = K.variable(3)
v2 = K.variable(4)
l1 = [r1,r2]
l2 = [v1,v2]
我正在尝试计算这两个张量的 MSE。我正在做的是:
res = []
for i in range(len(l1)):
res.append(K.square(l1[i] - l2[i]))
return sum(res)/len(res)
但我认为这段代码是大量的。有没有更有效,更优雅的方法来做到这一点?谢谢。
慕妹3242003
相关分类