torch.mm, torch.matmul 和torch.mul的区别: RuntimeError: 2D tensors expected, got 4D

点乘:torch.mul

torch.mul(a, b)是矩阵a和b点乘,a和b的维度必须相等,比如a的维度是(1, 2),b的维度是(1, 2),返回的仍是(1, 2)的矩阵

 

矩阵乘法:torch.mm和torch.mul

torch.mm(a, b)是矩阵a和b矩阵相乘,比如a的维度是(1, 2),b的维度是(2, 3),返回的就是(1, 3)的矩阵

torch.matmul则可以用于两个图片矩阵相乘,如a的维度是[bs,3,128,256]; b的维度是[bs, 3, 256,512], 则torch.mm(a, b)就会报错;但torch.matmul(a,b)就不会报错!

如下所示:

(Pdb) TL = torch.matmul(T,L.view(T.shape[0],T.shape[1],-1,1))
(Pdb) TL = torch.mm(T,L.view(T.shape[0],T.shape[1],-1,1))
*** RuntimeError: 2D tensors expected, got 4D, 4D tensors at /tmp/pip-req-build-ocx5vxk7/aten/src/THC/generic/THCTensorMathBlas.cu:282

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

R.X. NLOS

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值