[pytorch] in-place-operation

In-place-operation : 값을 변경할 떄 copy와 같은 함수를 사용하지 않고도 바로 변경해버리는 함수 예를 들어, a=3 b=5 c = a + b 라고 가정하자 기존에는 a+b를 진행해도 a값은 3이고 b의 값은 5이다. 그러나 여기서 ' a.add_(b)'로 작성한다면 (a+b의 의미임) a의 결과값은 8이 된다. import torch a = torch.tensor(3) b = torch.tensor(5) c = a + b # c = tensor(8) # a = tensor(3), b = tensor(5) a.add_(b) print(a) # a = tensor(8), b = tensor(5) 기존 파이썬에서는 이와 같은 연산을 진행하려면 copy와 같은 것을 사용해야 하는데, 이를 대신해서 값을 바로 바꿀 수 있는 함수라고 생각하면 되겠다.

장점 : 메모리를 아낄 수 있다. c를 담을 그릇을 만들 필요가 없음! 단점 : 기존에 a가 가지던 값을 기억해야할 경우에는...

요청 처리 중입니다...

[pytorch] in-place-operation

등록된 다른 글