pytorch matmul和mm和bmm区别

pytorch中matmul和mm和bmm区别

• matmul
• mm
• bmm
• 结论

先看下官网上对这三个函数的介绍。

matmul

mm

bmm

顾名思义, 就是两个batch矩阵乘法.

结论

从官方文档可以看出,

• mm只能进行矩阵乘法,也就是输入的两个tensor维度只能是$(n\times m)$(n×m)和$(m\times p)$(m×p)
• bmm是两个三维张量相乘, 两个tensor维度是$(b\times m\times p)$(b×m×p)和$(b\times n\times p)$(b×n×p), 第一维b代表batch_size
• matmul可以进行张量乘法, 输入可以是高维.