操作数不能与形状一起广播 (5,2) (1,5)

Question

我想将这两个 np.array 相乘，我相信它应该返回一个 (5,2) np.array，但它会引发错误。

a = np.array([1, 2, 3, 4, 5])
b = np.arange(10)
b = b.reshape((5,2))
print(a.shape, b.shape)
print(b * a)

Answer 1

如果你想在每一列上将 a 和 b 元素相乘，你需要添加

a = a.reshape((5,1))