pytorch各种乘法,mm, matmul, dot, @, *, mul, multiply

  1. torch.mm
    线代的矩阵乘法,要求输入都是矩阵

  2. torch.matmul
    注意:torch.mm和torch.matmul不等价
    根据输入不同执行不同的操作:

  • 输入都是二维矩阵,矩阵乘法,等同于torch.mm
  • 输入都是一维向量,计算向量内积,等同于torch.dot
  • 第一个参数是向量,第二个是矩阵,则将第一个参数变成(1,n)的矩阵,再执行矩阵乘法
  • 第一个参数是矩阵,第二个是向量,执行矩阵向量乘法,等同于torch.mv
  • 两个都是高维张量,自己看文档去
  1. torch.dot
    向量点积(内积),输入必须都是一维的。向量点积计算公式:
    a=(a1,a2,a3)
    b=(b1,b2,b3)
    ab=a1b1+a2b2+a3b3

因此向量内积是个标量

  1. torch.mul
    按元素相乘,element-wise的乘法,也叫哈达玛积

  2. torch.multiply
    torch.mul的别称

  3. *
    torch.mul的简写

  4. @
    torch.matmul的简写(注意不是torch.mat的简写)

  5. torch.outer
    向量外积,输入向量维度分别为n和m,则输出(n, m)

posted @   王冰冰  阅读(316)  评论(4编辑  收藏  举报
相关博文:
点击右上角即可分享
微信分享提示
💬
评论
📌
收藏
💗
关注
👍
推荐
🚀
回顶
收起