【pytorch】张量求导

笔者看到了这篇文章,可以很好的解释张量的求导问题:

看到了上面这张图,可以说很好的表示了前向和反向的过程了。

补充几个细节

之前看李沐的d2l,一直不懂为什么矩阵计算时的一些奇奇怪怪的规定,比如为什么一个行向量对另一个行向量作微分的结果是每一行都对第二个向量的每个元素都进行偏微分,然后拼成一个矩阵。按照作者提到的,首先将y中的每个元素都对向量进行微分,然后再拼成一个矩阵。

另外一个就是:为什么两个向量微分之后的形状是两个向量的形状直接拼起来,从文中作者的解释也可以看出,每个元素都对第二个向量作微分,形状和第二个向量相同,而每个结果都会作为第一个向量的元素,因为最后一维往往就是元素,所以效果就相当于两个向量拼起来。

作者没更矩阵部分的内容,好可惜哦。

相关推荐
Upsy-Daisy几秒前
Hermes Agent 学习笔记 01:一个会记忆、会学习、能长期运行的 AI Agent
人工智能·笔记·学习
小雨下雨的雨几秒前
五子棋AI在鸿蒙PC Electron上的实现的原理与实践
人工智能·游戏·华为·electron·harmonyos·鸿蒙
AI科技星1 分钟前
基于奇合数边界的离散解析数论与双螺旋宇大统一体系(中英文双语纯净终稿)
人工智能·线性代数·架构·概率论·学习方法
Swift社区2 分钟前
异构协同,算力重构:CPU+GPU架构下的AI推理优化
人工智能·重构·架构
调试优选官3 分钟前
2026上海AI搜索GEO排名优化:技术路径与服务能力解析
人工智能·ai·geo·上海
AI分享猿6 分钟前
编程教育的新篇章:AI工具如何改变教学方式
人工智能
Tiansan66668 分钟前
AI问答推广:郑州本地企业如何用技术提升40%转化率
人工智能·ai问答推广郑州本地
玩转单片机与嵌入式10 分钟前
AI 推理会不会堵住实时任务?MCU 上跑模型时,RTOS 和 DMA 该怎么配合?
人工智能·单片机·嵌入式硬件
张彦峰ZYF11 分钟前
LangGraph 条件边:让 AI Agent 学会“做选择”
人工智能·大模型·langgraph
ZFSS11 分钟前
BYOK(自带密钥)使用指南
运维·服务器·前端·人工智能·midjourney