我有一个多维数组 a 其形状为(32,3,5,5)和一个数组 v
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)