pytorch matmul和mm和bmm区别

pytorch中matmul和mm和bmm区别 matmul mm bmm 结论 先看下官网上对这三个函数的介绍。 matmul mm bmm 顾名思义, 就是两个batch矩阵乘法. 结论 从官方文档可以看出, mm只能进行矩阵乘法,也就是输入的两个tensor维度只能是 ( n × m ) (n\times m) (n×m)和 ( m × p ) (m\times p) (m×p) bmm是两

>>阅读原文<<