目录
- 1. pytorch的数学运算包括
- 2. 基本运算
- 3. matmul
- 4. power sqrt rsqrt
- 5. exp log
- 6. 近似值
- 7. clamp
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,那么分别最小值和最大值不受限