机器学习-Pytorch基础

  1. Numpy和Pytorch可以相互转换,前者CPU上,后者GPU上,都是对矩阵进行运算,Pytorch的基本单位是张量。
  2. torch 可以初始化全为0、全为1、符合正态分布的矩阵
  3. 确定性初始化
  • torch.tensor()
  • torch.arrange()
  • torch.linspace()
  • torch.logspace()
  • ones/zeros/eye/full
  1. 随机性初始化
  • torch.Tensor()
  • torch.rand()
  • torch.rand_like()
  • torch.randint()
  • torch.randn()
  1. 直接点×关闭jupyter不正确,应该再启动一个命令行,进入打开jupyter的环境,打jupyter notebook stop
  2. torch可以和python列表一样切片
  3. 神经网络输入一维,图片是三维,所以需要维度变换,view()和reshape()都可以维度变换,后者鲁棒性更强。reshape可以多层嵌套,前者不行。化成一维可以偷懒,用-1。
  4. 增加新的维度:unsqueeze(),增加一个括号
  5. 维度缩减:squeeze()维度值为1才能删,因为维度值大于1删除会丢数据,减少个括号。
  6. 维度扩展:expand(),repeat()
  7. 维度调整:transpose(),permute()前者只能调两个维度,后者可以调整多个维度。某个算法可能只能用最后一个维度,所以需要进行维度的调整,如transformer()
  8. 广播机制:相当于先进行了unsqueeze再进行expand。小维度向大维度进行广播时,得先匹配,得先小维度和大维度相同或者小维度为1才能进行广播。
  9. 根据文档进行编程的思想!不用背API,知道这个函数的功能是什么,用到的时候去查文档。
  10. 数学能力是科研能力的上限,如果是工作的话数学要求没那么高。
  11. 张量的拼接 cat()和stack(),前者只能是需要拼接的维度不同,其他得相同;后者所有维度都必须相同。
  12. 张量的拆分split()和chunk(),前者是按间隔拆分,后者是按数量拆分
  13. 取整floor(),ceil(),round()
  14. 裁数trunc(),frac()
  15. 计算mean,max,sum,prod,min
  16. a.argmax(),a.argmin()最大值索引和最小值索引
  17. eq(),equal()前者更精细,每个位置比对,后者是整体比对。

博客1
博客2
博客3

文档1

相关推荐
ZJU_统一阿萨姆3 小时前
【算子开发】矩阵乘法(GEMM)入门与共享内存优化
人工智能·线性代数·矩阵·系统架构
AI码农小姐姐3 小时前
AI漫剧用什么软件制作?知漫剧对比即梦/豆包/可灵怎么选?
人工智能
YH55269843 小时前
GPT‑5.6 Sol 原本支持 1M 上下文,Codex 现已放开此前限制,如何看待这次调整?
java·jvm·人工智能·gpt·算法·chatgpt
ZYJCSZKJ3 小时前
AI数字人实时交互系统的工程架构与多方言适配实践
人工智能·架构·交互·ai数字人直播系统
2601_965958463 小时前
口腔黏膜脱皮超2周未愈建议及时就医
人工智能·python
智购科技智能售货柜3 小时前
2026自动售货机整机可靠性测试:从高低温交变到EMC电磁兼容的认证工程实践~YH
运维·服务器·数据库·人工智能·物联网
“初生”4 小时前
用 Codex 做一致性 AI 动画:5 步工作流,角色不再漂移
人工智能·ai·chatgpt
AI_小站4 小时前
刚面完百度的 Agent 开发岗,我才发现:世界就是个巨大的草台班子
java·开发语言·人工智能·spring·百度·langchain
随风而飘1864 小时前
KEITHLEY吉时利 2400 数字源表
人工智能·功能测试·科技·测试工具
HyperAI超神经4 小时前
【Triton 教程】triton_language.fdiv
人工智能·深度学习·triton