[PyTorch] 텐서 간의 나눗셈 연산: torch.div() 설명

소개

  torch.div는 PyTorch에서 두 텐서 간의 요소별(element-wise) 나눗셈을 수행하는 함수입니다. 이 함수는 동일한 크기의 텐서끼리 나누거나, 스칼라 값을 텐서의 각 요소에 나눌 때 사용됩니다. torch.div는 벡터, 행렬, 다차원 텐서 등 다양한 차원의 텐서에 적용될 수 있으며, 딥러닝 모델에서 연산을 간편하게 수행하는 데 도움을 줍니다.

 

기본 사용법

상세 설명

• torch.div(tensor1, tensor2)는 tensor1을 tensor2로 요소별로 나누는 연산을 수행합니다.
• 동일한 크기의 텐서에 대해 연산이 가능하며, 브로드캐스팅을 통해 크기가 다른 텐서 간의 연산도 지원합니다.
• 스칼라 나눗셈
  
  • 텐서의 각 요소를 스칼라 값으로 나누려면 torch.div(tensor, value)를 사용할 수 있습니다. 이 경우 텐서의 모든 요소가 지정된 값으로 나뉩니다.
  • 이는 일정한 값을 모든 요소에 나누어야 하는 경우에 유용합니다.

예시 설명

• 첫 번째 예시에서 torch.div(tensor1, tensor2)는 [10.0, 20.0, 30.0]을 [2.0, 5.0, 10.0]으로 나누어 [5.0, 4.0, 3.0]을 생성합니다.
• 두 번째 예시에서는 스칼라 값 10으로 텐서 [10.0, 20.0, 30.0]의 각 요소를 나누어 [1.0, 2.0, 3.0]을 생성합니다.
• 세 번째 예시에서는 두 텐서 [10.0, 20.0, 30.0]과 [2.0, 4.0, 6.0]을 나누어 [5.0, 5.0, 5.0]을 생성합니다.

import torch

# 두 텐서 생성
tensor1 = torch.tensor([10.0, 20.0, 30.0])
tensor2 = torch.tensor([2.0, 5.0, 10.0])

# 텐서 나눗셈
result = torch.div(tensor1, tensor2)
print(result)
# 출력: tensor([5.0, 4.0, 3.0])

# 스칼라 값을 텐서에 나누기
tensor = torch.tensor([10.0, 20.0, 30.0])
result = torch.div(tensor, 10)
print(result)
# 출력: tensor([1.0, 2.0, 3.0])

# 두 텐서의 나눗셈과 추가 스칼라 값 곱하기
tensor1 = torch.tensor([10.0, 20.0, 30.0])
tensor2 = torch.tensor([2.0, 4.0, 6.0])
result = torch.div(tensor1, tensor2)
print(result)
# 출력: tensor([5.0, 5.0, 5.0])

 

라이센스

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

 

관련 내용