본문 바로가기
AI/Pytorch

torch.detach (and clone)

by Mujae98 2024. 8. 11.

torch.detach는 말 그대로 떼어내는 것이다.

어떻게 보면 두 가지 기능을 하는건데

복제와 연결을 끊는, 뗴어내는(detach)하는 것이다.

 

x = torch.tesnor([1,2,3,4])

z = x.clone(), z = x.detach() 같은데

detach의 경우에는 계산 그래프에서 떼어내는 역할도 한다.

아래의 그림이 이를 잘 표현해서 가져왔다.

 

 

이러면 backpropagation할 때 더이상 layer1까지 가지 않는다.

'AI > Pytorch' 카테고리의 다른 글

nn.Paramter()  (0) 2024.11.18
torch.rand()  (0) 2024.08.11