Numpy 中的 3-D 矩阵乘法

Question

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

bob.shape 给出 (2,4)

tim.shape 给出 (7,4)

这段代码给出了一个形状为(2,7,4)

的3维矩阵

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

它给出了我想要的输出。但是，我想知道在 numpy 中是否有更多的 elegant/faster 方法来执行此操作，而不是我必须遍历 bob

Answer 1

见Python Broadcasting

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