mean_norm_sq = 0
for n in range(1000):
    a = np.random.normal(0, 1, size=(5, 5, 5, 3))
s = a.sum(axis=(0, 1, 2))
mean_norm_sq += np.sum(s ** 2)
mean_norm_sq / 1000
Векторная сумма многомерных массивов в numpy