以下内容继续学习Pytorch的数学运算。
1. 矩阵乘法
(1)mm(a,b) 只适用于2D
(2)a@b
(3)matmul(a,b) 可用于多维
2. 次方运算
也能利用次方来开方:
3. 幂次方
4. 近似方法
其中floor()向下取整,ceil()向上取整,trunc()取整数部分,frac()取小数部分
round()函数代表四舍五入。
5.clamp裁剪功能
max()求最大,median()求中位数,clamp(10)表示小于10的都用10来代替。clamp(0, 10)表示大于10的用10来表示,小于0的都用0来表示。
6. 统计属性
其中prod()表示累乘,argmax()返回最大值的索引,argmin()返回最小值的索引。