python - Numpy:如何按元素乘以形状 (n,1) 和 (n,) 的两个向量？

Question

如果两个向量具有相同的形状，比如都是 (n,1) 或都是 (n,)，则两​​个向量的逐元素乘法没有问题。但是，如果一个向量的形状为 (n,1) 而另一个向量的形状为 (n,)，则 * 运算符会返回一些有趣的东西。

a = np.ones((3,1))
b = np.ones((3,))
print a * b

生成的 nxn 矩阵包含 A_{i,j}=a_i*b_j。

我怎样才能对 a 和 b 进行逐元素乘法？

Answer 1

以使其形状匹配的方式对向量进行切片:

a[:, 0] * b

或

a * b[:, None]