如何將一個向量與一個矩陣元素明確地在numpy中相乘？

Question

我有一個多維數組的形狀是（32,3,5,5）和形狀爲（32，）的數組。我怎麼能乘以（我，3,5,5）與（i，）每個我使用numpy以外的for循環？

Answer 1

隨着a和b爲兩個數組，幾種方法可以建議 -

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