本站原创文章,转载请说明来自《老饼讲解-深度学习》www.bbbdata.com
pytorch求梯度时会在forward时生成运算图,并在backward时销毁运算图,即动态运算图机制
本文用例子展示自动梯度的运算图,并进一步用代码例子讲解动态图机制
通过本文,可以了解什么是自动梯度的运算图,以及pytorch中的动态图机制指的是什么
本节讲解pytorch的运态算运算是什么意思
自动求梯度产生的运算图是指类似如下的运算图![]()
它是为了实现自动求梯度而将目标函数拆解为基本初等函数从而得到的运算图
pytorch中的运算图是动态运算图,这里的"动态"是指:
👉1.在forward过程中,构建运算图
👉2.在backward时,销毁运算图
示例如下:
好了,上述就是pytorch的动态运算图了~
End