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

为君一笑 提交于 2020-04-23 04:49:25
  • 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机制

 

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!