zl程序教程

您现在的位置是:首页 >  其他

当前栏目

torch中的几种乘法:torch.mm, torch.mul, torch.matmul &&矩阵相乘和矩阵点乘

amp 几种 矩阵 乘法 MM torch 相乘
2023-09-11 14:22:51 时间

一、点乘

点乘都是broadcast的,可以用torch.mul(a, b)实现,也可以直接用*实现。

矩阵点乘,就是矩阵各个对应元素相乘,要求矩阵必须维数相等,即MxN维矩阵乘以MxN维矩阵 。


二、矩阵乘

当a, b维度不一致时,会自动填充到相同维度相点乘。

矩阵相乘有torch.mm和torch.matmul两个函数。其中前一个是针对二维矩阵,后一个是高维。当torch.mm用于大于二维时将报错。

矩阵相乘,就是矩阵的乘法操作,要求左矩阵的列和右矩阵的行数要一样,即MN维矩阵乘以和NY维矩阵