zl程序教程

您现在的位置是:首页 >  其它

当前栏目

torch.mul() 和 torch.mm() 区别【矩阵a和b对应位相乘/矩阵相乘】

区别 矩阵 对应 MM torch 相乘
2023-09-14 09:06:09 时间

torch.mul(a, b)是矩阵a和b对应位相乘,a和b的维度必须相等,比如a的维度是(1, 2),b的维度是(1, 2),返回的仍是(1, 2)的矩阵
torch.mm(a, b)是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b的维度是(2, 3),返回的就是(1, 3)的矩阵


torch.mul() 和 torch.mm() 的区别_Real_Brilliant的博客-CSDN博客_torch.mm()