PyTorch矩阵乘法

总结：pytorch和numpy中，星号*都表示矩阵点对点相乘；matmul都表示矩阵乘法。

===========================================================================

PyTorch中，a.mul(b)表示矩阵a和b点对点乘（即相同位置的元素相乘），‘*’也表示矩阵点对点相乘。a.mm(b)表示矩阵乘法。另外torch.matmul(a, b)以及a@b也都可以表示a和b的矩阵乘积。

对于a.mul(b)，b也可以是标量：

对于.mul()和‘*’，维数不同也可以相乘：

 

posted @   Picassooo  阅读(2008)  评论(0)  编辑  收藏  举报
相关博文：
· pytorch和Numpy 的广播机制高效计算矩阵之间两两距离
· pytorch多维张量相乘和广播机制示例
· pytorch各种乘法，mm, matmul, dot, @, *, mul, multiply
· PyTorch 中的乘法：mul()、multiply()、matmul()、mm()、mv()、dot()
· torch中的几种乘法：torch.mm, torch.mul, torch.matmul &&矩阵相乘和矩阵点乘
阅读排行：
· Manus爆火，是硬核还是营销？
· 终于写完轮子一部分：tcp代理 了，记录一下
· 别再用vector＜bool＞了！Google高级工程师：这可能是STL最大的设计失误
· 震惊！C++程序真的从main开始吗？99%的程序员都答错了
· 单元测试从入门到精通
点击右上角即可分享
微信分享提示