PyTorch에서 텐서 요소의 로그 계산하기: torch.log() 사용법 완벽 정리

PyTorch에서 텐서(tensor) 요소의 로그값을 계산하려면 torch.log() 메서드를 사용합니다. 이 메서드는 입력 텐서의 각 요소에 대한 자연로그(natural logarithm) 값을 가진 새로운 텐서를 반환합니다. 텐서를 입력 매개변수로 받아 연산 결과를 텐서 형태로 출력하는 간단한 구조입니다.

사용 단계

• 필요한 라이브러리 임포트: 아래 모든 예제에서는 torch 라이브러리가 필요합니다. 사전에 설치가 완료되어 있는지 확인하세요.

• 텐서 생성 및 출력: 로그를 계산할 텐서를 만들고 화면에 출력합니다.

• torch.log(input) 호출: input으로 텐서를 전달하면, 해당 텐서의 각 요소에 대한 자연로그 값을 담은 새로운 텐서를 반환받습니다.

• 결과 출력: 원본 텐서 요소들의 자연로그 값이 저장된 결과 텐서를 출력하여 확인합니다.

예제 1: 1차원 텐서의 자연로그 계산

다음 파이썬 프로그램은 PyTorch 텐서의 각 요소에 대해 자연로그를 계산하는 기본적인 방법을 보여줍니다.

# 필요한 라이브러리 임포트
import torch

# 텐서 생성
t = torch.Tensor([2.3,3,2.3,4,3.4])

# 생성된 텐서 출력
print("Original tensor:\n", t)

# 텐서 요소들의 로그 계산
log = torch.log(t)

# 계산된 로그값 출력
print("Logarithm of Elements:\n", log)

출력 결과

Original tensor:
    tensor([2.3000, 3.0000, 2.3000, 4.0000, 3.4000])
Logrithm of Elements:
    tensor([0.8329, 1.0986, 0.8329, 1.3863, 1.2238])

예제 2: 2차원 텐서의 자연로그 계산

다음 프로그램은 2차원(2D) 텐서에 대해서도 동일하게 torch.log()를 적용할 수 있음을 보여줍니다.

# 필요한 라이브러리 임포트
import torch

# 크기 3x4인 난수 텐서 생성
t = torch.rand(3,4)

# 생성된 텐서 출력
print("Original tensor:\n", t)

# 텐서 요소들의 로그 계산
log = torch.log(t)

# 계산된 로그값 출력
print("Logarithm of Elements:\n", log)

출력 결과

Original tensor:
tensor([[0.1245, 0.0448, 0.1176, 0.7607],
        [0.7415, 0.7738, 0.0694, 0.6983],
        [0.8371, 0.6169, 0.3858, 0.8027]])
Logarithm of Elements:
tensor([[-2.0837, -3.1048, -2.1405, -0.2735],
        [-0.2990, -0.2565, -2.6676, -0.3591],
        [-0.1778, -0.4830, -0.9524, -0.2198]])