- 加 / 减 / 乘 / 除
- 矩阵相乘
- 幂函数
- 开方 / 开方的倒数
- 近似值
1.加 / 减 / 乘 / 除
2. 矩阵相乘
- Torch.mm
- only for 2d
- Torch.matmul
-
@
2d tensor 矩阵相乘
3. 幂函数
-
Power
-
Exp log
4. 开方 / 开方的倒数
5. 近似值
- .floor() .ceil()
向下取整;向上取整
*.trunc() .frac()
只保留整数;只保留小数 -
.round()
四舍五入
6. 梯度裁剪
clamp(min)
clamp
(min, max)
网友评论