Question

我有两个名为numpy和A的2D B数组，A为M x N且B为{ {1}}。我的问题是，我希望将M x n的每一行的与B的相应行相乘，并创建一个3D矩阵{{1其大小为A，不使用C - 循环。

例如，如果M x n x N是：

for

和A是

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

然后得到的乘法B看起来像

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

我想要实现的目标是否明确，是否可以在没有任何C = A x B循环的情况下进行？最好的，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]]])

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

2 个答案: