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擴展成和另外一個input維度同樣的Tensor,再點乘。須要input的最後一維維數相同,例如[4]→[5,3,4],不然報錯。

 

broadcast機制

 

 

相關文章
相關標籤/搜索