【笔记】【Pytorch】关于torch.matmul和torch.bmm的输出tensor数值不一致问题

最新推荐文章于 2025-03-08 17:47:44 发布

Laox1ao

最新推荐文章于 2025-03-08 17:47:44 发布

阅读量2.9w

点赞数 3

分类专栏：神经网络学习记录文章标签： pytorch matmul bmm

本文链接：https://blog.youkuaiyun.com/Laox1ao/article/details/79159303

版权

博客指出torch.matmul和torch.bmm在处理batch矩阵乘法时，虽然打印出的数值看似相同，但在高精度比较下存在差异，尤其是在小数点后第七位开始。这提醒我们在进行相等判断时要注意精度问题，因为即便是同一操作，不同的输入顺序也可能导致不一致的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

发现

对于torch.matmul和torch.bmm，都能实现对于batch的矩阵乘法：

a = torch.rand((2,3,10))
b = torch.rand((2,2,10))
### matmal()
res1 = torch.matmul(a,b.transpose(1,2))
print res1 
"""
...
[torch.FloatTensor of size 2x3x2]
"""
### bmm()
res2 = torch.bmm(a,b.transpose(1,2))
print res2 
"""
...
[torch.FloatTensor of size 2x3x2]
"""