[PyTorch] 차원을 줄여라: torch.squeeze() 함수 사용법 이해하기

소개

  torch.squeeze 함수는 PyTorch에서 텐서의 크기가 1인 차원을 제거하여 텐서를 간소화하는 함수입니다. 이 함수는 주로 데이터 전처리 과정에서 불필요한 차원을 제거하여 연산 효율을 높이기 위해 사용됩니다.

 

기본 사용법

상세 설명

• torch.squeeze 함수의 기본 구문은 torch.squeeze(input, dim=None)입니다.
  
  • input: 차원을 제거할 입력 텐서.
  • dim: 제거할 차원을 지정합니다. 이 인수가 주어지지 않으면, 크기가 1인 모든 차원이 제거됩니다.
• torch.squeeze 함수는 입력 텐서에서 크기가 1인 차원을 제거하여 새로운 텐서를 반환합니다.

예시 설명

• torch.squeeze(tensor)는 3차원 텐서 [[[1], [2], [3]]]에서 크기가 1인 첫 번째 차원을 제거하여 1차원 텐서 [1, 2, 3]를 생성합니다.
• torch.squeeze(tensor, dim=0)는 3차원 텐서 [[[1, 2, 3]], [[4, 5, 6]]]에서 첫 번째 차원을 제거하여 2차원 텐서 [[1, 2, 3], [4, 5, 6]]를 생성합니다.
• torch.squeeze(tensor_gpu)는 GPU 장치에서 3차원 텐서 [[[1], [2], [3]]]의 크기가 1인 차원을 제거하여 1차원 텐서 [1, 2, 3]를 생성합니다.

# 기본
import torch

tensor = torch.tensor([[[1], [2], [3]]])
squeezed = torch.squeeze(tensor)
print(squeezed)
# 출력: tensor([1, 2, 3])

# 다양한 크기와 데이터 타입
import torch

# 특정 차원만 제거하기
tensor = torch.tensor([[[1, 2, 3]], [[4, 5, 6]]])
squeezed = torch.squeeze(tensor, dim=0)
print(squeezed)
# 출력: tensor([[1, 2, 3],
#              [4, 5, 6]])

# GPU에서 텐서 차원 제거
tensor_gpu = torch.tensor([[[1], [2], [3]]], device=torch.device('cuda'))
squeezed_gpu = torch.squeeze(tensor_gpu)
print(squeezed_gpu)
# 출력: tensor([1, 2, 3], device='cuda:0')

 

라이센스

  PyTorch의 표준 라이브러리와 내장 함수들은 BSD-style license 하에 배포됩니다. 이 라이센스는 자유 소프트웨어 라이센스로, 상업적 사용을 포함한 거의 모든 용도로 사용이 가능합니다. 라이센스와 저작권 정보는 PyTorch의 공식 GitHub 리포지토리에서 확인할 수 있습니다.

 

관련 내용

  [PyTorch] 차원을 추가하라: torch.unsqueeze() 함수 사용법 이해하기

  [PyTorch] 텐서 펼치기: torch.flatten() 사용 가이드

  [PyTorch] 차원을 추가하여 텐서 쌓기: torch.stack() 함수 활용법

  [PyTorch] 텐서 이어 붙이기: torch.cat() 함수 활용법