将两个2D numpy数组乘以3D数组

Question

我有两个二维numpy数组叫A和B,这里A是M x N和B是M x n.我的问题是我希望将每行的每个元素B与相应的行相乘,A并创建一个C大小的3D矩阵M x n x N,而不使用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的情况下进行？最好的,tingis

Answer 1

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