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是兩

>>阅读原文<<