一、點乘
點乘都是broadcast的,可以用torch.mul(a, b)實現,也可以直接用*實現。
矩陣點乘,就是矩陣各個對應元素相乘,要求矩陣必須維數相等,即MxN維矩陣乘以MxN維矩陣 。
二、矩陣乘
當a, b維度不一致時,會自動填充到相同維度相點乘。
矩陣相乘有torch.mm和torch.matmul兩個函數。其中前一個是針對二維矩陣,后一個是高維。當torch.mm用於大於二維時將報錯。
矩陣相乘,就是矩陣的乘法操作,要求左矩陣的列和右矩陣的行數要一樣,即MN維矩陣乘以和NY維矩陣