numpy中的元素计算

对于两个数组中的每个元素,我要计算均方误差,即

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

我试着按顺序计算这个元素,但没有找到有效的方法。