前言：当我们在训练网络的时候可能希望保持一部分的网络参数不变，只对其中一部分的参数进行调整；或者值训练部分分支网络，并不让其梯度对主网络的梯度造成影响，这时候我们就需要使用detach()函数来切断一些分支的反向传播。

1 tensor.detach()

返回一个新的tensor，从当前计算图中分离下来的，但是仍指向原变量的存放位置,不同之处只是requires_grad为false，得到的这个tensor永远不需要计算其梯度，不具有grad。

即使之后重新将它的requires_grad置为true,它也不会具有梯度grad

这样我们就会继续使用这个新的tensor进行计算，后面当我们进行反向传播时，到该调用detach()的tensor就会停止，不能再继续向前进行传播

注意：使用detach返回的tensor和原始的tensor共同一个内存，即一个修改另一个也会跟着改变。

比如正常的例子是：

import torcha = torch.tensor([1, 2, 3.], requires_grad=True)
print(a.grad)  # None
out = a.sigmoid()
out.sum().backward()  # tensor([0.1966, 0.1050, 0.0452])
print(a.grad)

1.1 当使用detach()分离tensor但是没有更改这个tensor时，并不会影响backward():

import torcha = torch.tensor([1, 2, 3.], requires_grad=True)
print(a.grad)  # None
out = a.sigmoid()
print(out)  # tensor([0.7311, 0.8808, 0.9526], grad_fn=<SigmoidBackward>)# 添加detach(),c的requires_grad为False
c = out.detach()
print(c)  # tensor([0.7311, 0.8808, 0.9526])# 这个时候没有对c进行更改，所以不会影响backward()
out.sum().backward()
print(a.grad)  # tensor([0.1966, 0.1050, 0.0452])

从上可见tensor c是由out分离得到的，但是我也没有去改变这个c，这个时候依然对原来的out求导是不会有错误的，即

c,out之间的区别是c是没有梯度的，out是有梯度的,但是需要注意的是下面两种情况是汇报错的

1.2 当使用detach()分离tensor，然后用这个分离出来的tensor去求导数，会影响backward()，会出现错误

import torcha = torch.tensor([1, 2, 3.], requires_grad=True)
print(a.grad)  # None
out = a.sigmoid()
print(out)  # tensor([0.7311, 0.8808, 0.9526], grad_fn=<SigmoidBackward>)# 添加detach(),c的requires_grad为False
c = out.detach()
print(c)  # tensor([0.7311, 0.8808, 0.9526])# 使用新生成的Variable进行反向传播
c.sum().backward()
print(a.grad)'''
Traceback (most recent call last):File "/opt/data/private/lhl/SSL/uncertainty_maps/main_detach.py", line 13, in <module>c.sum().backward()File "/root/anaconda3/envs/py379/lib/python3.7/site-packages/torch/tensor.py", line 221, in backwardtorch.autograd.backward(self, gradient, retain_graph, create_graph)File "/root/anaconda3/envs/py379/lib/python3.7/site-packages/torch/autograd/__init__.py", line 132, in backwardallow_unreachable=True)  # allow_unreachable flag
RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn
'''

1.3 当使用detach()分离tensor并且更改这个tensor时，即使再对原来的out求导数，会影响backward()，会出现错误

如果此时对c进行了更改，这个更改会被autograd追踪，在对out.sum()进行backward()时也会报错，因为此时的值进行backward()得到的梯度是错误的：

import torcha = torch.tensor([1, 2, 3.], requires_grad=True)
print(a.grad)  # None
out = a.sigmoid()
print(out)  # tensor([0.7311, 0.8808, 0.9526], grad_fn=<SigmoidBackward>)# 添加detach(),c的requires_grad为False
c = out.detach()
print(c)  # tensor([0.7311, 0.8808, 0.9526])
c.zero_()  # 使用inplace函数对其进行修改# 这个时候会发现对c进行更改，会影响backward
c.sum().backward()
print(a.grad)'''
Traceback (most recent call last):File "/opt/data/private/lhl/SSL/uncertainty_maps/main_detach.py", line 14, in <module>c.sum().backward()File "/root/anaconda3/envs/py379/lib/python3.7/site-packages/torch/tensor.py", line 221, in backwardtorch.autograd.backward(self, gradient, retain_graph, create_graph)File "/root/anaconda3/envs/py379/lib/python3.7/site-packages/torch/autograd/__init__.py", line 132, in backwardallow_unreachable=True)  # allow_unreachable flag
RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn
'''

2 tensor.detach_()

将一个tensor从创建它的图中分离，并把它设置成叶子tensor

其实就相当于变量之间的关系本来是x -> m -> y,这里的叶子tensor是x，但是这个时候对m进行了m.detach_()操作,其实就是进行了两个操作：

将m的grad_fn的值设置为None,这样m就不会再与前一个节点x关联，这里的关系就会变成x, m -> y,此时的m就变成了叶子结点
然后会将m的requires_grad设置为False，这样对y进行backward()时就不会求m的梯度

总结：其实detach()和detach_()很像，两个的区别就是detach_()是对本身的更改，detach()则是生成了一个新的tensor

比如x -> m -> y中如果对m进行detach()，后面如果反悔想还是对原来的计算图进行操作还是可以的

但是如果是进行了detach_()，那么原来的计算图也发生了变化，就不能反悔了

3 tensor.clone()

clone(memory_format=torch.preserve_format)→ Tensor

返回tensor的拷贝，返回的新tensor和原来的tensor具有同样的大小和数据类型。

原tensor的requires_grad=True
clone()返回的tensor是中间节点，梯度会流向原tensor，即返回的tensor的梯度会叠加在原tensor上

参考资料

https://blog.csdn.net/qq_27825451/article/details/95498211
https://zhuanlan.zhihu.com/p/389738863

pytorch的两个函数 tensor.detach()，tensor.detach_()，tensor.clone() 的作用和区别相关推荐

详解PyTorch中的copy_()函数、detach()函数、detach_()函数和clone()函数
参考链接: copy_(src, non_blocking=False) → Tensor 参考链接: detach() 参考链接: detach_() 参考链接: clone() → Tensor ...
pytorch：.detach()、.detach_()的作用和区别
文章目录 pytorch的两个函数:.detach()..detach_()的作用和区别一.torch.detach() 二.tensor.detach_() 补充:requires_grad.gr ...
PyTorch中 tensor.detach() 和 tensor.data 的区别
原文链接请参考:https://dreamhomes.top/posts/201906081516/ PyTorch0.4中,.data 仍保留,但建议使用 .detach(), 区别在于 .data ...
pytorch: Variable detach 与 detach_
pytorch 的 Variable 对象中有两个方法,detach和 detach_ 本文主要介绍这两个方法的效果和能用这两个方法干什么. detach 官方文档中,对这个方法是这么介绍的. 返回 ...
Pytorch 中的detach 和detach_
本文转自 http://blog.csdn.net/u012436149/article/details/76714349 pytorch 的 Variable 对象中有两个方法,detach和 de ...
【深度学习】pytorch自动求导机制的理解 | tensor.backward() 反向传播 | tensor.detach()梯度截断函数 | with torch.no_grad()函数
提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档文章目录前言一.pytorch里自动求导的基础概念 1.1.自动求导 requires_grad=True 1.2.求导 requ ...
pytorch 中的Tensor.detach介绍
detach的作用 Tensor.detach() 的作用是阻断反向梯度传播,当我们再训练网络的时候可能希望保持一部分的网络参数不变,只对其中一部分的参数进行调整:或者值训练部分分支网络,并不让其梯度 ...
pytorch tensor.detach
import torch as t x = t.ones(1, requires_grad=True) x.requires_grad #True y = t.ones(1, requires_gra ...
PyTorch 笔记（02）— 常用创建 Tensor 方法（torch.Tensor、ones、zeros、eye、arange、linspace、rand、randn、new）
1. Tensor 概念分类 PyTorch 中的张量(Tensor)类似 NumPy 中的 ndarrays,之所以称之为 Tensor 的另一个原因是它可以运行在 GPU 中,以加速运算. 1.1 ...

pytorch的两个函数 tensor.detach()，tensor.detach_()，tensor.clone() 的作用和区别