1.数学操作
- add/minus/multiply/divide
- matmul
- pow
- sqrt/rsqrt
- round
2.加减乘除
加法
矩阵乘法
- torch.mm
- 只适用于2d
- torch.matmul
- @
例子
二维以上的tensor进行矩阵相乘
pow/sqrt
rsqrt()函数是计算机中用来计算算数平方根倒数的函数
微信扫一扫
加法
例子
二维以上的tensor进行矩阵相乘
rsqrt()函数是计算机中用来计算算数平方根倒数的函数
相关推荐