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

相关推荐
田里的水稻3 小时前
PyTorch_Dataloader\TensorBoard---数据加载与可视化
人工智能·pytorch·python·神经网络·机器学习
田里的水稻4 小时前
FA_CNN卷积神经网络三---非线性激活函数PyTorch
人工智能·pytorch·神经网络·机器学习·cnn
打工仔折腾 AI6 小时前
从零开始大模型微调:基于 PyTorch 与 ChatGLM 的 LoRA 实践指南
人工智能·pytorch·后端·python·深度学习·性能优化·ai agent 实战
燃烧的枫8 小时前
Windows 下 YOLO26 的 TensorRT C++ 部署
人工智能·pytorch·windows·计算机视觉·tensorrt·cuda·yolo26
ZhengEnCi13 小时前
用 YOLOv8n + 自研二分类模型做课堂行为识别:从 9000 张标注数据到 Flask 全栈落地
pytorch·python·计算机视觉
承渊政道2 天前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(从ChatGLM本地化到LoRA、PEFT的高级微调)
android·人工智能·pytorch·chatglm·本地化部署·大模型开发与微调
Smoothcloud润云2 天前
GPU算力平台推荐:新手租GPU时,镜像环境和显存应该怎么选?
大数据·人工智能·pytorch·分布式·深度学习·gpu算力
田里的水稻3 天前
PyTorch_TensorBoard展示
人工智能·pytorch·python·深度学习·神经网络
想七想八不如114083 天前
元学习简单入门
人工智能·pytorch·python·深度学习·学习·机器学习
田里的水稻3 天前
PyTorch_torchvision---数据集下载\TensorBoard可视化
人工智能·pytorch·python·深度学习·机器学习