当前位置: 首页> 教育> 锐评 > 【pytorch09】数学运算

【pytorch09】数学运算

时间:2025/7/11 11:11:55来源:https://blog.csdn.net/qq_45291280/article/details/140020700 浏览次数:0次

1.数学操作

  • add/minus/multiply/divide
  • matmul
  • pow
  • sqrt/rsqrt
  • round

2.加减乘除

加法
在这里插入图片描述

矩阵乘法

  • torch.mm
    • 只适用于2d
  • torch.matmul
  • @

要分清楚是矩阵元素相乘,还是矩阵相乘

在这里插入图片描述
例子
在这里插入图片描述

x一共有4张照片,每张照片打平成784的向量,希望降维得到[4,512],所以我们会构造一个w矩阵负责把784降维成512,因此w应该定义为[784,512],才会得到[4,512],但是上述代码并没有将w定义成[784,512],因为pytorch默认w的维度是[channel_out,channel_in]即[输出的维度,输入的维度],所以要进行一个转置,如果w是高纬度的要使用transport要交换维度

二维以上的tensor进行矩阵相乘

只取最后的2维进行运算,前面2维保持不变
其实就是支持多个矩阵对并行相乘

在这里插入图片描述

pow/sqrt

rsqrt()函数是计算机中用来计算算数平方根倒数的函数
在这里插入图片描述
在这里插入图片描述

round四舍五入

在这里插入图片描述

clamp裁剪

在这里插入图片描述

关键字:【pytorch09】数学运算

版权声明:

本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。

我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com

责任编辑: