[pytorch] unsqueeze 함수, 차원 늘리기

이번 글에서는 torch tensor의 차원을 늘리는 방법중에 하나인 unsqueeze함수에 대해 알아보겠다. 이 글을 보기 전에 차원의 숫자 1인 차원을 없애는 방법인 squeeze를 봐고 와도 좋다. ([pytorch] squeeze사용법: 텐서의 차원 제거하기)안봐도 된다.

 

Unsqueeze 함수 사용법

import torch 
tensor = torch.rand(10, 64, 128)
tensor = tensor.unsqueeze(dim=2) #[10, 64, 128] -> [10,64,1,128]

위와 같이 내가 차원을 증가시키고 싶은 차원에 해당하는 인덱스를 입력하면 차원이 증가하는 것을 확인할 수 있다.