python - Numpy 中的 3-D 矩阵乘法

Question

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

bob.shape给(2,4)

tim.shape给(7,4)

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

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

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

标签： pythonpython-3.xnumpymatrixnumpy-ndarray

Answer 1

请参阅Python 广播

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