python_package/numpy

numpy - squeeze() / np.squeeze()

Xenrose 2024. 9. 29. 13:32

아래 내용은 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