首页 > 解决方案 > Numpy ValueError:操作数无法与形状一起广播(152,1)(151,1)

问题描述

我有两个形状相同的 numpy 数组 (152, 1),我将它们放入我制作的均方误差函数中,如下所示:

return np.sum(y - y_hat, axis=1, keepdims=True)**2/len(y_hat)

但由于某种原因,我收到了错误:

ValueError: operands could not be broadcast together with shapes (152,1) (151,1)

我知道错误是由 y - y_hat 引起的,因为我试图打印出来看看发生了什么,但我得到了同样的错误。我不知道问题可能是什么,因为这是一个完全合法的矩阵运算,并且两个 numpy 数组的形状都是相同的。任何帮助将不胜感激,谢谢。

编辑:我刚刚意识到 y_hat 是一个短值,我认为它们是相同的。对不起。

标签: pythonpython-3.xnumpyarray-broadcasting

解决方案


推荐阅读