【pytorch】张量求导

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

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

补充几个细节

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

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

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

相关推荐
xzl042 分钟前
当使用 AutoTokenizer 加载百川(Baichuan)模型时出现 BaiChuanTokenizer 相关报错
人工智能·pytorch·python
yangshuo12813 分钟前
心灵宝石MCP部署完全指南:AI IDE积分零损耗的实现方案
ide·人工智能·microsoft
L.fountain4 分钟前
图像自回归生成(Auto-regressive image generation)实战学习(三)
人工智能·深度学习·学习·回归
咕噜企业分发小米5 分钟前
腾讯云知识图谱实体链接的准确率如何评估?
人工智能·算法·机器学习
前端程序猿之路6 分钟前
简易版AI知识助手项目 - 构建个人文档智能问答系统
前端·人工智能·python·ai·语言模型·deepseek·rag agent
geneculture6 分钟前
融智学:重构认知与实践的智慧体系
大数据·人工智能·融智学的重要应用·信智序位·人类智力·融智时代(杂志)
Chukai1237 分钟前
第3章:基于LlamaIndex+Ollama+ChromaDB搭建本地简单RAG问答系统
开发语言·人工智能·python·rag·rag问答系统
智源研究院官方账号8 分钟前
智源数字孪生心脏升级:全自动跨尺度心脏毒性推演,构建“可计算”的药物研发未来
人工智能
私人珍藏库9 分钟前
[吾爱大神原创工具] PythonEnvManager - Python 环境管理工具 [更新自定义扫描路径]
开发语言·python
前端程序猿之路11 分钟前
AI大模型应用开发之Prompt 工程
人工智能·python·深度学习·机器学习·prompt·ai编程·提示词