본문 바로가기

PyTorch5

[PyTorch] torch.tensor.detach() https://pytorch.org/docs/stable/generated/torch.Tensor.detach.html torch.Tensor.detach — PyTorch 2.3 documentationShortcutspytorch.org  파이토치에서 Tensor 객체의 detach() 메소드는 현재 Tensor 객체와 동일한 데이터를 가지지만 연산 그래프(Computational Graph) 에서 분리된 새로운 Tensor 객체를 생성한다. 즉, 이 메소드는 일반적으로 Tensor 객체를 다른 Tensor 객체로 변환하고자 할 때 사용된다.  자세히 설명해보면, 주어진 Tensor 객체에 대한 연산의 결과로 생성된 새로운 Tensor 객체가 있을 때, 이 새로운 Tensor 객체를 사용하여 추가적인.. 2024. 7. 17.
[PyTorch] torch.scatter_ 알아보자 torch.scatter_() https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html#torch.Tensor.scatter_ Parameter dim : scatter 할 기준이 되는 축. '0'이면 행 방향, '1'이면 열 방향 index (LongTensor) : 흩뿌릴 element들의 index. 즉, 어떤 숫자를 어떤 규칙으로 옮길지 결정하는 tensor src : 어떤 숫자들이 옮겨지는지 그 후보를 담은 소스 tensor torch.scatter는 scatter_의 out of place 버전이다. 흔히 inplace = False 옵션을 쓰는 것과 동일하다. 따라서 scatter로 tensor를 조작하고 다시 변수에 할당해주.. 2023. 12. 14.
[PyTorch] make_grid() 사용하는 방법 torchvision.utils.make_grid() (https://pytorch.org/vision/stable/generated/torchvision.utils.make_grid.html) make_grid — Torchvision 0.16 documentation Shortcuts pytorch.org 이미지 텐서들을 모아서 grid 형태로 만들어주는 함수로, 이미지 시각화에 유용하다. 공식문서를 참고해서 설명해보면 다음과 같다. Parameters tensor (Tensor or list) : 4D mini-batch Tensor (Batch, Channel, Height, Width) 또는 같은 크기의 이미지 리스트 nrow (int, optional) : 한 행에 표시될 이미지의 개수. 최종.. 2023. 12. 13.