如何将矩阵与向量相乘，使得每一行都与这个向量按元素相乘

Question

我有一个尺寸为1682 x 36的矩阵和一个尺寸为1 x 36的向量。我要将每一行元素与这个向量1 × 36相乘。

一个例子:

a = np.array([[1, 2],
[2, 1]])
b = np.array([1, 2])
result= np.array([[1, 4],
[2, 2]])

Answer 1

尝试一个简单的a*b，应该根据您在问题中发布的内容工作-

作为一个例子-

a = np.random.random((1686,36))
b = np.random.random((1,36))
output = a*b
output.shape

(1686, 36)

如果第二个是(36,)形状，则尝试-

a * b[None,:]

如果你遇到形状问题，这应该会添加一个额外的轴。