반응형
Pytorch 텐서 차원변경
-
[PyTorch Study]Tensor 자료형 사이즈 및 차원 변경2) Tech 2020. 9. 23. 09:04
PyTorch Study 세번째 시간입니다!! 지난 게시물에 이어 Pytorch Tensor 자료형 응용 2번째 시간입니다. 1. Tensor 자료형 사이즈 변경 # pytorch reshape, view 사용법 import torch t1 = torch.tensor([1,2,3,4,5,6]) print( t1, '\n') t2 = t1.view(2, 3) print( t2, '\n') print( t1.reshape(2,3), '\n') 위 코드의 출력을 보시면 reshape, view 의 역할이 감이 오실겁니다! - torch.tensor().reshape, view : Tensor 자료형의 사이즈 및 차원 변경시 사용 굉장히 많이 사용되므로 사용법을 잘알아 두셔야합니다. 이번에는 차원변경 예시를 함..