您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

numpy dot()和Python 3.5+矩阵乘法之间的区别@

numpy dot()和Python 3.5+矩阵乘法之间的区别@

@运营商称阵列的__matmul__方法,而不是dot。此方法在API中也作为函数存在np.matmul

>>> a = np.random.rand(8,13,13)
>>> b = np.random.rand(8,13,13)
>>> np.matmul(a, b).shape
(8, 13, 13)

从文档中:

matmul区别于dot两个重要方面。

最后一点很明确,当传递3D(或更高维)数组时,dotmatmul方法的行为会有所不同。从文档中引用更多内容

对于matmul

如果任一自变量的值为ND,N> 2,则将其视为驻留在最后两个索引中的矩阵的堆栈,并进行相应广播。

对于np.dot

对于2-D数组,它等效于矩阵乘法,对于1-D数组,其等效于向量的内积(无复共轭)。 对于N维,它是a的最后一个轴和b的倒数第二个轴的和积

python 2022/1/1 18:50:01 有384人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶