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()

您正在计算a-b，将其平方，然后分别对每一行求和，然后求和。