아래 내용은 numpy 1.26 버전으로 작성됨.

0. 바로 사용하기

numpy.squeeze(a,
			  axis=None)

• a: input data
• axis: 축소할 차원의 축

---

1. 기본형

numpy.squeeze(a,
			  axis=None)

---

2. 기능

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

---

3. 파라미터

a

• squeeze를 진행할 input data

axis

• squeeze를 진행할 data의 축
  (torch.squeeze() 의 dim 파라미터와 같은 역할)
  단, axis 인자값으로 받은 축 중에서 차원이 1이 아닐 경우 오류 출력
  

---

ref

• https://numpy.org/doc/stable/reference/generated/numpy.squeeze.html