假設(shè)有兩個張量列表：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)但我認為這段代碼是大量的。有沒有更有效，更優(yōu)雅的方法來做到這一點？謝謝。
1 回答
慕妹3242003
TA貢獻1824條經(jīng)驗 獲得超6個贊
我認為你應該能夠簡單地做到：
return K.mean(K.square(K.stack(l1) - K.stack(l2)))
請注意，我假設(shè)列表中的所有張量都具有相同的形狀（就像在您的代碼段中一樣，否則也會失?。?。
添加回答
舉報
0/150
提交
取消