numpy中的元素智能計算

Question

對於兩個數組中的每個元素，我想計算均方誤差，即

sum((a - b)^2)) / num of rows

例如，如果我有

aa = np.array([[1,2,3,4],[0,0,0,0]])
bb = np.array([[1,2,3,0],[1,1,1,1]])

MSE將變為(16 + 4) / 2 = 10

我試圖以元素為單位進行計算，但是沒有找到有效地實現此目標的好方法。

Answer 1

我不確定您的有效含義是什么，但這應該可以工作：

((aa-bb)**2).sum(axis=1).mean()

您正在計算ab，對它進行平方，然后分別對每一行求和，然后求和。