- 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机制