【pytorch】张量求导

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

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

补充几个细节

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

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

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

相关推荐
python在学ing3 小时前
前端-CSS学习笔记
前端·css·python·学习
烈风逍遥3 小时前
基于 Vue 3+Spring Boot 构建 RAG 智能知识库
人工智能
机器之心3 小时前
1/10成本、Opus 4.7级表现,Cursor甩出了性价比之王Composer 2.5
人工智能·openai
机器之心3 小时前
从卖token到卖结果,这些公司开始让AI背KPI了
人工智能·openai
效能革命笔记3 小时前
AI模型平台选型指南:开源生态与国产算力如何优选
人工智能·开源
机器之心3 小时前
H100去哪儿了?
人工智能·openai
多年小白3 小时前
兆易创新分析
大数据·人工智能·ai·金融·区块链
小领航3 小时前
构建 MySQL MCP Server
人工智能·node.js
paperClub4 小时前
AACR 2026 · AI诊断:深度学习在肿瘤早期检测中的应用
人工智能·深度学习
碳基硅坊4 小时前
使用RAGFlow搭建本地知识库
人工智能·知识库·rag·ragflow