아래 내용은 pytorch 2.1 버전으로 작성됨.

0. 바로 사용하기

torch.squeeze(input)

• input: 크기가 1인 차원을 제거할 tensor

---

1. 기본형

torch.squeeze(input, 
			  dim=None)

---

2. 기능

• 입력 받은 array에서 크기가 1인 차원을 전부 제거하고 객체를 반환함.
  (shape의 크기가 1인 차원을 지우는 것)
  
  

---

3. 파라미터

input

• 차원을 제거할 array 객체

dim

• dim = integer: 정수를

• dim = tuple: 튜플을

입력 받아 해당 정수에 해당하는 차원의 크기가 1일 경우 제거

---

ref

• https://pytorch.org/docs/2.1/generated/torch.squeeze.html#torch-squeeze