1.数学操作

• add/minus/multiply/divide
• matmul
• pow
• sqrt/rsqrt
• round

2.加减乘除

加法

矩阵乘法

• torch.mm
  • 只适用于2d
• torch.matmul
• @

例子

二维以上的tensor进行矩阵相乘

pow/sqrt

rsqrt()函数是计算机中用来计算算数平方根倒数的函数

round四舍五入

clamp裁剪