如何在numpy中将向量乘以数组/矩阵元素的方式?

问题描述 投票:4回答:1

我有一个形状为(32,3,5,5)的多维数组a和一个形状为(32,)的数组v。如何使用除for循环以外的numpy为每个i乘以(i,3,5,5)与(i,)?

python numpy numpy-ndarray numpy-broadcasting elementwise-operations
1个回答
2
投票

[av作为两个数组,几乎没有建议方法-

a*v[:,None,None,None]
a*v.reshape(-1, *[1]*3)
(a.T * v).T
np.einsum('i...,i->i...', a, v)
© www.soinside.com 2019 - 2024. All rights reserved.