美文网首页
(九)基本运算

(九)基本运算

作者: zelda2333 | 来源:发表于2020-02-12 10:33 被阅读0次
    • 加 / 减 / 乘 / 除
    • 矩阵相乘
    • 幂函数
    • 开方 / 开方的倒数
    • 近似值

    1.加 / 减 / 乘 / 除

    2. 矩阵相乘

    • Torch.mm
      • only for 2d
    • Torch.matmul
    • @


    2d tensor 矩阵相乘


    3. 幂函数

    • Power
    • Exp log

    4. 开方 / 开方的倒数

    5. 近似值

    • .floor() .ceil()
      向下取整;向上取整
      *.trunc() .frac()
      只保留整数;只保留小数
    • .round()
      四舍五入


    6. 梯度裁剪

    clamp(min)
    clamp
    (min, max)


    相关文章

      网友评论

          本文标题:(九)基本运算

          本文链接:https://www.haomeiwen.com/subject/jhudactx.html