pytorch #딥러닝 #tensormanipulation #viewvsreshape #python코딩
-
PyTorch의 view()와 reshape() 비교Pytorch 2025. 3. 6. 12:52
PyTorch에서 텐서의 크기를 변경하는 두 가지 주요 방법은 view()와 reshape()이다. 두 함수 모두 기존 텐서의 크기를 변경할 수 있지만, 내부적으로 동작 방식에 차이가 있다. 본 글에서는 두 함수의 차이점과 언제 각각을 사용해야 하는지 설명한다.1. view() 함수view()는 기존 텐서의 메모리 공유를 유지하면서 크기를 변경한다. 즉, 원본 텐서와 새로운 텐서는 같은 메모리를 가리키므로, 하나를 변경하면 다른 하나에도 영향을 미친다. 하지만 view()를 사용하려면 텐서가 연속적인(Contiguous) 메모리 레이아웃을 가져야 한다.예제 코드:import torchtensor = torch.arange(6).reshape(2, 3)view_tensor = tensor.view(3, 2..