1.数学操作
- add/minus/multiply/divide
- matmul
- pow
- sqrt/rsqrt
- round
2.加减乘除
加法

矩阵乘法
- torch.mm
- 只适用于2d
- torch.matmul
- @

例子

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

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


round四舍五入

clamp裁剪

微信扫一扫
加法


例子

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

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




相关推荐