pytorch torch.detach函数(返回一个新的`Variable`,从当前图中分离下来的)
Posted Dontla
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pytorch torch.detach函数(返回一个新的`Variable`,从当前图中分离下来的)相关的知识,希望对你有一定的参考价值。
detach()[source]
Returns a new Variable, detached from the current graph. 返回一个新的Variable
,从当前图中分离下来的。
返回的Variable requires_grad=False
,如果输入 volatile=True
,那么返回的Variable
volatile=True
。
注意:
返回的Variable
和原始的Variable
公用同一个data tensor
。in-place
修改会在两个Variable
上同时体现(因为它们共享data tensor
),可能会导致错误。
参考文章:https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-autograd/#detachsource
以上是关于pytorch torch.detach函数(返回一个新的`Variable`,从当前图中分离下来的)的主要内容,如果未能解决你的问题,请参考以下文章
从 pytorch 数据集返回索引:更改 __getitem__ 的函数导致元类冲突