Numpy中的3D矩阵乘法

Question

我必须在Numpy Python 3.x中将两个二维矩阵bob和tim相乘

bob.shape给出(2,4)

tim.shape给出(7,4)

这段代码给出了形状为(2,7,4)的3-D矩阵

np.array([foo*tim for foo in bob])

它给出我想要的输出。但是，我想知道是否有一种更优雅/更快的方式以numpy的方式执行此操作，而不是我不得不遍历bob

Answer 1

请参见Python Broadcasting

bob.reshape((2, 1, 4)) * tim.reshape((1, 7, 4))