torch.mul, mm, matmul, bmm, broadcast乘法機制

torch.mm只能讓兩個二維tensor作矩陣乘法 torch.mul作element-wise的矩陣點乘,維數不限,可以矩陣乘標量 torch.bmm作batch單位的矩陣乘法,維度只能爲3。當第0維維數不等時報錯,但可用matmul相乘 torch.matmul 正常的矩陣乘法運算,兩個輸入必須都是Tensor torch.mul有broadcast機制,可以把其中一個input擴展成和另一
相關文章
相關標籤/搜索