pytorch数学运算

1. pytorch的数学运算包括

▪Add/minus/multiply/divide
▪Matmul
▪Pow
▪Sqrt/rsqrt
▪Round

2. 基本运算

+、-、*、/ 也可以使用函数add sub mul div

3. matmul

矩阵相乘有三种方式：
1、torch.mm 针对2d矩阵
2、torch.matmul 多维矩阵
3、@

注意：矩阵只有当左边矩阵的列数等于右边矩阵的行数时,它们才可以相乘

4. power sqrt rsqrt

power-几次幂
sqrt-平方根
rsqrt-平方根的倒数

5. exp log

exp 指数，log对数

6. 近似值

.floor() 取整数的最小值
.ceil() 取整数的最大值
.round() 取整
.trunc() 整数部分
.frac() 小数部分

7. clamp

torch.clamp(input, min=None, max=None, *, out=None) → Tensor
把输入值控制在[min, max]之间，如果min=None或者max=None，那么分别最小值和最大值不受限