地心的PyTorch学习(二)

以下内容继续学习Pytorch的数学运算。

1. 矩阵乘法

(1)mm(a,b) 只适用于2D

地心的PyTorch学习(二)_第1张图片

(2)a@b

地心的PyTorch学习(二)_第2张图片 

(3)matmul(a,b) 可用于多维

地心的PyTorch学习(二)_第3张图片

 地心的PyTorch学习(二)_第4张图片

2. 次方运算 

地心的PyTorch学习(二)_第5张图片

也能利用次方来开方:

地心的PyTorch学习(二)_第6张图片

 3. 幂次方

 地心的PyTorch学习(二)_第7张图片

 地心的PyTorch学习(二)_第8张图片

4. 近似方法

地心的PyTorch学习(二)_第9张图片 

 其中floor()向下取整,ceil()向上取整,trunc()取整数部分,frac()取小数部分

地心的PyTorch学习(二)_第10张图片

round()函数代表四舍五入。

 5.clamp裁剪功能

地心的PyTorch学习(二)_第11张图片

 max()求最大,median()求中位数,clamp(10)表示小于10的都用10来代替。clamp(0, 10)表示大于10的用10来表示,小于0的都用0来表示。

6. 统计属性

地心的PyTorch学习(二)_第12张图片 

其中prod()表示累乘,argmax()返回最大值的索引,argmin()返回最小值的索引。

地心的PyTorch学习(二)_第13张图片

 

你可能感兴趣的:(Pytorch,pytorch,学习,深度学习)