Поэлементный расчет в 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, возводите в квадрат, затем суммируете каждую строку отдельно, а затем усредняете суммы.