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

相关推荐
玫幽倩9 小时前
2026第二届湾区杯网络安全大赛决赛(AI专项赛道静态题wp)
pytorch·python·ai·agent·ctf·rag·湾区杯
论文复现现场11 小时前
课程作业要跑 PyTorch 训练,学校机房不够用去哪租?云 GPU 选型、环境迁移与防丢数据指南
人工智能·pytorch·深度学习·云计算·gpu·cuda
崖边看雾12 小时前
深度学习——PyTorch 实现 CBOW 词向量模型:从数据预处理到训练与保存
pytorch·深度学习
ysu_03141 天前
边界条件的硬约束与软约束——从试函数构造到谱分析诊断
人工智能·pytorch·深度学习·物理信息神经网络·pinns·边界条件·ntk
会飞锦鲤1 天前
基于 Mask R-CNN 的药片缺陷检测系统
人工智能·pytorch·python·神经网络·resnet-50
马剑威(威哥爱编程)1 天前
【共创稿事节】HarmonyOS 7 Agent A2A 实战:让日程智能体和打车智能体自己谈成一单
pytorch·深度学习·harmonyos
内蒙深海大鲨鱼1 天前
8-13课程归纳
pytorch
派大_星1 天前
基于 PyTorch 实现 CBOW 词向量模型
pytorch
JarmanYuo1 天前
YOLO 涨点研究(十二):具身 CV 进阶篇——Sim2Real 域随机化与真机部署
人工智能·pytorch·python·yolo·计算机视觉
马剑威(威哥爱编程)2 天前
【共创稿事节】HarmonyOS 7 应用 Skill 化实战:从“被打开“到“被调用“,把功能递进系统意图分发池
pytorch·深度学习·harmonyos