【龙龙老师 pytorch】⑥ 数学运算
(1)add sub mul div / 加减乘除
乘法:对应位置元素相乘
(2)矩阵相乘
- Torch.mm 仅适用于2dim
- torch.matmul 推荐
- @
(3)power 次方
power(X)= ** X
(4)exp log
(5)Approximation - floor 下取整
- ceil 上取整
- round 四舍五入
- trunc 取整数部分
- frac 取小数部分
(6)clamp
多用于梯度裁剪
(1)add sub mul div / 加减乘除
乘法:对应位置元素相乘
(2)矩阵相乘