- torch.mm只能讓兩個二維tensor做矩陣乘法
- torch.mul做element-wise的矩陣點乘,維數不限,能夠矩陣乘標量
- torch.bmm做batch單位的矩陣乘法,維度只能爲3。當第0維維數不等時報錯,但可用matmul相乘
- torch.matmul 正常的矩陣乘法運算,兩個輸入必須都是Tensor
- torch.mul有broadcast機制,能夠把其中一個input擴展成和另外一個input維度同樣的Tensor,再點乘。須要input的最後一維維數相同,例如[4]→[5,3,4],不然報錯。
broadcast機制