pytorch各种乘法,mm, matmul, dot, @, *, mul, multiply
-
torch.mm
线代的矩阵乘法,要求输入都是矩阵 -
torch.matmul
注意:torch.mm和torch.matmul不等价
根据输入不同执行不同的操作:
- 输入都是二维矩阵,矩阵乘法,等同于torch.mm
- 输入都是一维向量,计算向量内积,等同于torch.dot
- 第一个参数是向量,第二个是矩阵,则将第一个参数变成(1,n)的矩阵,再执行矩阵乘法
- 第一个参数是矩阵,第二个是向量,执行矩阵向量乘法,等同于torch.mv
- 两个都是高维张量,自己看文档去
- torch.dot
向量点积(内积),输入必须都是一维的。向量点积计算公式:
则
因此向量内积是个标量
-
torch.mul
按元素相乘,element-wise的乘法,也叫哈达玛积 -
torch.multiply
torch.mul的别称 -
*
torch.mul的简写 -
@
torch.matmul的简写(注意不是torch.mat的简写) -
torch.outer
向量外积,输入向量维度分别为n和m,则输出(n, m)
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步