將兩個2D numpy數組乘以3D數組

Question

我有兩個名為A和B 2D numpy數組，其中A是M x N ， B是M xn 。 我的問題是我希望將每行 B 每個元素與A相應行相乘，並創建一個大小為M xnx N的3D矩陣C ，而不使用for -loops。

例如，如果A是：

A = np.array([[1, 2, 3],
              [4, 5, 6]])

而B是

B = np.array([[1, 2],
              [3, 4]])

然后得到的乘法C = A x B看起來像

C = [
     [[1, 2],
      [12, 16]],
     [[2, 4],
      [15, 20]],
     [[3, 6],
      [18, 24]]
     ]

是否清楚我想要實現的目標，是否可以在沒有任何for -loops的情況下for ？ 最好的，tingis

Answer 1

C=np.einsum('ij,ik->jik',A,B)

Answer 2

可以通過在每個數組中創建一個新軸並轉置修改后的A ：

A[np.newaxis,...].T * B[np.newaxis,...]

贈送：

array([[[ 1,  2],
        [12, 16]],

       [[ 2,  4],
        [15, 20]],

       [[ 3,  6],
        [18, 24]]])