我有一个形状为(32,3,5,5)的多维数组a和一个形状为(32,)的数组v。如何使用除for循环以外的numpy为每个i乘以(i,3,5,5)与(i,)?
a
v
[a和v作为两个数组,几乎没有建议方法-
a*v[:,None,None,None] a*v.reshape(-1, *[1]*3) (a.T * v).T np.einsum('i...,i->i...', a, v)