【pytorch】张量求导

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

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

补充几个细节

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

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

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

相关推荐
twilight_469几秒前
机器学习与模式识别——机器学习中的搜索算法
人工智能·python·机器学习
冰西瓜6009 分钟前
深度学习的数学原理(十)—— 权重如何自发分工
人工智能·深度学习·计算机视觉
niuniudengdeng18 分钟前
基于时序上下文编码的端到端无文本依赖语音分词模型
人工智能·数学·算法·概率论
Jia ming43 分钟前
《智能法官软件项目》—罪名初判模块
python·教学·案例·智能法官
Jia ming1 小时前
《智能法官软件项目》—法律文书生成模块
python·教学·案例·智能法官软件
曦月逸霜1 小时前
Python数据分析——个人笔记(持续更新中~)
python
海棠AI实验室1 小时前
第六章 从“能用”到“能交付”的关键一刀:偏好对齐(Preference Alignment)数据工程
python·私有模型训练
百锦再1 小时前
Java多线程编程全面解析:从原理到实战
java·开发语言·python·spring·kafka·tomcat·maven
Soonyang Zhang1 小时前
flashinfer attention kernel分析
人工智能·算子·推理框架
林籁泉韵72 小时前
2026年GEO服务商推荐:覆盖多场景适配,助力企业AI时代增长
人工智能