python - numpy中不同形状的数组之间的乘法

Question

我是 Python 新手，我不知道如何在不同形状的数组之间执行乘法运算。我有两个不同的数组w，b这样：

W.shape = [32, 5, 20]
b.shape = [5,]

我想乘

W[:, i, :]*b[i]

对于每个 i 从 0 到 4。我该怎么做？提前致谢。

标签： pythonarraysnumpyvectormultiplication

Answer 1

您可以添加一个新轴，b使其与“内部数组”行相乘W，即第二个轴：

W * b[:,None]