如何在 numpy 中將向量乘以數組/矩陣元素？

Question

我有一個形狀為 (32,3,5,5) 的多維數組a和一個形狀為 (32,) 的數組v 。 對於除 for 循環之外的每個 i，我如何使用 numpy 將 (i,3,5,5) 與 (i,) 相乘？

Answer 1

使用a和v作為兩個數組，可以建議幾種方法 -

a*v[:,None,None,None]
a*v.reshape(-1, *[1]*3)
(a.T * v).T
np.einsum('i...,i->i...', a, v)