您可以使用：
mse = ((A - B)**2).mean(axis=ax)
要么
mse = (np.square(A - B)).mean(axis=ax)
- 与
ax=0
平均值沿着行进行的，对于每一列，返回一个数组 - 与
ax=1
平均值沿着列进行的，对于每一行，返回一个数组 - 与
ax=None
平均值沿着阵列进行逐元素，返回一个标量值
您可以使用：
mse = ((A - B)**2).mean(axis=ax)
要么
mse = (np.square(A - B)).mean(axis=ax)
ax=0平均值沿着行进行的，对于每一列，返回一个数组
ax=1平均值沿着列进行的，对于每一行，返回一个数组
ax=None平均值沿着阵列进行逐元素，返回一个标量值
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
