torch中的幾種乘法:torch.mm, torch.mul, torch.matmul &&矩陣相乘和矩陣點乘


一、點乘

點乘都是broadcast的,可以用torch.mul(a, b)實現,也可以直接用*實現。

矩陣點乘,就是矩陣各個對應元素相乘,要求矩陣必須維數相等,即MxN維矩陣乘以MxN維矩陣 。


二、矩陣乘

當a, b維度不一致時,會自動填充到相同維度相點乘。

矩陣相乘有torch.mm和torch.matmul兩個函數。其中前一個是針對二維矩陣,后一個是高維。當torch.mm用於大於二維時將報錯。

矩陣相乘,就是矩陣的乘法操作,要求左矩陣的列和右矩陣的行數要一樣,即MN維矩陣乘以和NY維矩陣


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM