pytorch矩陣相乘matmul,mm

在這裏插入圖片描述

注意:torch.mm和轉置(.t())只是適用於2維矩陣,推薦使用torch.matmul(等價於@)

多維度數據的轉置使用的是transpose或者permute。

在這裏插入圖片描述
matmul:只取後兩個維度進行相乘。符合broadcast機制的可以相乘。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章