[Pytorch 용어 정리] Squeeze, Unsqueeze

* Squeeze함수 

 

- 차원이 1인 차원을 제거해 준다. 따로 차원을 설정하지 않으면 1인 차원을 모두 제거한다. 그리고 차원을 설정해 주면 그 차원만 제거한다.

 

* 주의할 점은 생각치도 못하게 batch가 1일때 batch차원도 없애버리는 불상사가 발생할 수 있다. 그래서 validation 단계에서 오류가 날 수 있기 때문에 주의해서 사용해야 한다.

 

* Unsqueeze함수

 

- squeeze함수의 반대로 1인 차원을 생성하는 함수이다. 그래서 어느 차원에서 1인 차원을 생성할지 꼭 지정해 주어야 한다. 

 

ex) x.unsqueeze(dim=1) # [3,20,128- ->[3,1,20,128]