Pytorch 기초 Tensor 크기 변경하기 차원 재구성 view() squeeze()

오늘은 Pytorch 기초 두번째 포스팅을 하려고 합니다. 저번 포스팅에서는 텐서를 선언하는 방법과 선언한 텐서에 계산 관련 함수를 적용하는 방법을 배웠습니다. 이번 포스팅에서는 텐서의 크기를 변경하는 함수들에 대해 다뤄봅니다. ** 본 포스팅은 pc 버전에 최적화 되어 있습니다. ** 텐서 크기 변경 함수 view() squeeze() unsqueeze() 타입캐스팅 - 자료형 변환 cat() stack() ones_like() / zeros_like() In-place → mul_() 1. 뷰 (view) 원소의 수를 유지하면서 텐서의 크기 변경 torch :: view = numpy :: reshape import numpy as np t = np.array([[[0,1,2], [3,4,5]], [[6,7,8], [9,10,11]]]) # 3차원 ft = torch.FloatTensor(t) # 3차원 텐서 ft.shape # torch.Size([2, 2, 3]) 면, 행,...

원문링크 : Pytorch 기초 Tensor 크기 변경하기 차원 재구성 view() squeeze()