【pytorch09】数学运算

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裁剪

相关推荐
程序员羽痕32 分钟前
基于 Django + PyTorch 的中文字体识别系统
pytorch·python·django
培之17 小时前
RTX 5090 安装 pytorch3d
人工智能·pytorch·python
延卿18 小时前
nnDetection:基于 PyTorch 的目标检测框架
人工智能·pytorch·目标检测
不懒不懒19 小时前
Windows 深度学习环境配置(CUDA12.8 + cuDNN9.x + PyTorch)最简避坑指南(2026 最新)
人工智能·pytorch·深度学习
丨白色风车丨20 小时前
PyTorch 迁移学习实战:ResNet18 实现 20 类食物图像分类(完整可运行代码)
pytorch·分类·迁移学习
七夜zippoe20 小时前
深入解析CANN仓库中的HCCL分布式通信库
pytorch·分布式·cann·hccl·通信库
All The Way North-1 天前
【TorchMetrics精通系列③】模型评估进阶:ClasswiseWrapper 实战与多分类细粒度指标深度解析
pytorch·模型评估·分类指标·包装器·评估指标·torchmetrics
不如语冰2 天前
AI大模型入门-模块导入import
数据结构·人工智能·pytorch·python
FriendshipT2 天前
Ultralytics:解读CBLinear模块
人工智能·pytorch·python·深度学习·目标检测
不如语冰2 天前
AI大模型入门-参数的传递
数据结构·人工智能·pytorch·python