【pytorch】张量求导

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

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

补充几个细节

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

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

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

相关推荐
zh路西法1 分钟前
【离策略强化学习】DDPG、TD3 与 SAC:核心公式推导与源码解析
python·强化学习·ddpg·td3·sb3·off-policy·sac
秋秋小事3 分钟前
python 数据契约层FastAPI+Pydantic(一)
python
AIFQuant3 分钟前
Python港股实时行情API接入:恒生股票数据获取与AH股溢价监控实战
开发语言·数据库·python·金融·实时数据api
小果冻。。4 分钟前
大模型生成原理与幻觉控制:从原理到实战
人工智能
沐风___4 分钟前
REA AI 逆向工程:没有源码,怎样查清软件的一个功能?
人工智能·机器学习
蒲公英eric5 分钟前
线性回归与正则化:从最小二乘到岭回归、LASSO
人工智能·算法·数学建模·线性回归·岭回归·最小二乘·lasso
ElectroComp8 分钟前
高功率 PoE 与 AI 算力电源双轮驱动,磁性元器件迎来技术升级新契机
人工智能
renhongxia110 分钟前
中美AI竞争下人工智能赋能大众生活
大数据·人工智能·生活
AI增长技术研究院10 分钟前
AI心智指数(绿雪智能)怎么用?AI可见性检测与GEO优化步骤
人工智能
朝阳资本论10 分钟前
小马智行杀入伦敦牵手Uber,中国Robotaxi出海怎么打?
python