pytorch编程入门

【介绍】pytorch的动态运算图

作者 : 老饼 发表日期 : 2023-07-28 10:45:06 更新日期 : 2025-05-30 04:56:51
本站原创文章,转载请说明来自《老饼讲解-深度学习》www.bbbdata.com



pytorch求梯度时会在forward时生成运算图,并在backward时销毁运算图,即动态运算图机制

本文用例子展示自动梯度的运算图,并进一步用代码例子讲解动态图机制

通过本文,可以了解什么是自动梯度的运算图,以及pytorch中的动态图机制指的是什么





一、 pytorch的动态运算图   



本节讲解pytorch的运态算运算是什么意思



0.1 运算图回顾


自动求梯度产生的运算图是指类似如下的运算图
运算图是什么 
它是为了实现自动求梯度而将目标函数拆解为基本初等函数从而得到的运算图



02.pytorch的动态运算图


pytorch中的运算图是动态运算图,这里的"动态"是指:
👉1.在forward过程中,构建运算图
👉2.在backward时,销毁运算图

示例如下:
动态运算图是什么






好了,上述就是pytorch的动态运算图了~







 End 






图标 评论
添加评论