[PyTorch] 텐서 요소의 합 계산: torch.sum() 설명

소개

  torch.sum은 PyTorch에서 텐서의 모든 요소에 대한 합을 계산하는 함수입니다. 이 함수는 텐서의 특정 차원(axis)에서 합계를 구할 수 있는 옵션을 제공하며, 다양한 차원의 텐서에 적용할 수 있습니다. torch.sum은 벡터, 행렬, 다차원 텐서의 요소를 모두 더하거나 특정 차원에서 합계를 구하는 데 유용합니다.

 

기본 사용법

상세 설명

• 전체 요소 합계
  
  • torch.sum(tensor)는 텐서의 모든 요소를 더하여 총합을 반환합니다.
  • 이 기능은 텐서의 전체 합계를 필요로 하는 다양한 계산에 유용합니다.
• 차원별 합계
  
  • torch.sum(tensor, dim=0)와 같이 dim 인수를 사용하면 특정 차원(axis)에서 합계를 계산할 수 있습니다.
  • 이 기능은 특히 텐서의 각 차원에서 합계를 구하여 데이터를 축소하거나 특정 차원에서의 통계를 구할 때 유용합니다.

예시 설명

• 첫 번째 예시에서 torch.sum(tensor)는 1D 텐서 [1, 2, 3, 4, 5]의 모든 요소를 더하여 15를 반환합니다.
• 두 번째 예시에서는 2D 텐서의 모든 요소의 합계는 21이며, dim=0과 dim=1을 사용하여 각각 열과 행의 합계를 구할 수 있습니다.

import torch

# 1D 텐서 생성
tensor = torch.tensor([1, 2, 3, 4, 5])

# 모든 요소의 합계 계산
result = torch.sum(tensor)
print(result)
# 출력: tensor(15)

# 2D 텐서 생성
tensor = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 모든 요소의 합계 계산
total_sum = torch.sum(tensor)
print(total_sum)
# 출력: tensor(21)

# 특정 차원에서의 합계 계산 (dim=0: 각 열의 합)
column_sum = torch.sum(tensor, dim=0)
print(column_sum)
# 출력: tensor([5, 7, 9])

# 특정 차원에서의 합계 계산 (dim=1: 각 행의 합)
row_sum = torch.sum(tensor, dim=1)
print(row_sum)
# 출력: tensor([ 6, 15])

 

라이센스

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

 

관련 내용