텐서를 사용하다보면 텐서의 모양을 바꿔야 하는 경우가 생긴다. torch 에서는 이를 위하여 squeeze, unsqueeze를 제공한다. squeeze는 차원을 줄일때 사용하며, unsqueeze는 차원을 늘릴때 사용한다. 주의 : squeeze는 차원이 1인 경우에 해당하는 모든 차원을 줄여준다. unsqueeze는 특정 위치에 차원이 1인 경우를 추가한다. - 여기서 특정 위치의 시작은 0부터 시작하며, 앞에서 부터 올라간다. x = torch.zeros(1,1,32,32) x.size() #torch.Size([1, 1, 32, 32]) x = torch.squeeze(x) x.size() #torch.Size([32, 32]) x = torch.unsqueeze(x,0) x.size() #torch.Size([1, 32, 32]) 만약 특정 위치가 1이 아닌 경우에 차원을 줄이려고 하면, 결과는 변하지 않는다. x = torch.zeros(1,32,32) x = torch....
![[pytorch] squeeze, unsqueeze](https://blogimgs.pstatic.net/nblog/mylog/post/og_default_image_160610.png)
#
pytorch
#
squeeze
#
unsqueeze
#
텐서
#
토치
#
파이토치
원문 링크 : [pytorch] squeeze, unsqueeze