다음은 난수 생성에 대한 실행 결과입니다.
7 0.5517938632336726 5.378512915003691 1
③ torch.squeeze()는 텐서의 차원을 줄이고자 할 때 사용합니다. 사용 방법은 다음과 같습니다.
import torch x = torch.FloatTensor([[1], [2]]) ------ (2×1) 크기의 2차원 텐서 생성 print(x) print(x.shape) print('--squeeze 적용--') print(x.squeeze()) ------ squeeze( )가 적용되어 1차원으로 축소 print(x.squeeze().shape)
다음은 squeeze()를 적용한 결과입니다.
tensor([[1.], [2.]]) torch.Size([2, 1]) --squeeze 적용-- tensor([1., 2.]) torch.Size([2])
squeeze()를 적용한 결과 (2×1) 텐서가 (2,) 크기를 갖는 1차원 벡터로 변경되었습니다.