0%

Pytorch之matmul函数

发表于 更新于 分类于 阅读次数： 阅读次数： Valine：
本文字数： 302 阅读时长 ≈ 1 分钟

写在前面：

参考文章：

  1. https://blog.csdn.net/ganxiwu9686/article/details/95204013

1. 矩阵相乘

从官方文档可以看出

  • mm只能进行矩阵乘法,也就是输入的两个tensor维度只能是（b m）, (m k) 得到（b * k）

  • bmm是两个三维张量相乘, 两个tensor维度是，（b m n）, (b n k) 得到（b m k）

  • matmul可以进行张量乘法, 输入可以是高维.

总结：
对位相乘用torch.mul；二维矩阵乘法用torch.mm；batch二维矩阵用torch.bmm；batch，广播用torch.matmul