python – Numpy中的三维矩阵乘法

发布时间：2020-12-16 22:23:51 所属栏目：Python 来源：网络整理

导读：我必须在Numpy Python 3.x中将两个2-D矩阵,bob和tim相乘 bob.shape给出(2,4) tim.shape给出(7,4) 这段代码给出了一个形状为(2,7,4)的三维矩阵 np.array([foo*tim for foo in bob]) 它给出了我想要的输出.但是,我想知道是否有一个更优雅/更快的方式来做这个在

我必须在Numpy Python 3.x中将两个2-D矩阵,bob和tim相乘

bob.shape给出(2,4)

tim.shape给出(7,4)

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

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

它给出了我想要的输出.但是,我想知道是否有一个更优雅/更快的方式来做这个在numpy而不是我不得不迭代bob

最佳答案

见Python Broadcasting

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!