矩陣的numpy元素乘法3d * 3d = 4d

Question

我有一個尺寸爲m * n * x的numpy ndarray A和尺寸爲m * n * y的另一個numpy ndarray B. 我想將它們相乘並以下面的方式獲得的尺寸M * N * X * Y的ndarray C：

for i in range(x): 
    for j in range(y): 
     C[:,:, i, j] = numpy.multiply(A[:, :, i], B[:, :, j]) 

Answer 1

C = A[:, :, :, numpy.newaxis] * B[:, :, numpy.newaxis, :] 

插入新的長度-1在A和B軸在適當的點'形狀和應用廣播增殖。