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扩展成和另一
相关文章
相关标签/搜索