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

相关推荐
阿图灵1 天前
Seq2Seq Transformer 中英翻译实战:从 GRU 到 Transformer,BLEU 0.225 → 0.307
pytorch·深度学习·gru·transformer·机器翻译·seq2seq
程序猿编码1 天前
不用PyTorch,我用C++手写了一个能认手写数字的神经网络
c++·pytorch·神经网络·大模型
129Lab1 天前
数据驱动电池SOH预测实战:用AI工具快速构建深度学习模型并调参
pytorch·深度学习·ai编程·数据驱动·特征工程·电池管理系统·电池健康状态
软糯小野猫1 天前
使用Pytorch从零开始构建StyleGAN
pytorch·深度学习·gan·图像生成·stylegan
hhzz2 天前
《深度学习框架PyTorch入门与实践》系列:11-实战猫狗大战之可复用的PyTorch项目架构
人工智能·pytorch·深度学习
玩三国杀玩的2 天前
Pytorch-c++-CUDA
c++·pytorch·python·深度学习
内蒙深海大鲨鱼2 天前
5.Introduction to PyTorch YouTube Series--visualable train
人工智能·pytorch·python
承渊政道2 天前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(从注意力到自回归生成:彻底理解Transformer解码器)
pytorch·回归·transformer·chatglm·注意力机制·解码器
盼小辉丶2 天前
PyTorch强化学习实战(22)——将强化学习应用于TextWorld互动小说游戏
pytorch·深度学习·强化学习
Uncommon.3 天前
使用Pytorch自动计算梯度
人工智能·pytorch·python