Computer >> Máy Tính >  >> Lập trình >> Python

Làm thế nào để tính Logarit của các phần tử của một tensor trong PyTorch?

Để tính logarit của các phần tử của tensor trong PyTorch, chúng tôi sử dụng torch.log () phương pháp. Nó trả về một tensor mới với các giá trị logarit tự nhiên của các phần tử của tensor đầu vào ban đầu. Nó nhận một tensor làm tham số đầu vào và đầu ra một tensor.

Các bước

  • Nhập thư viện được yêu cầu. Trong tất cả các ví dụ Python sau, thư viện Python bắt buộc là torch . Đảm bảo rằng bạn đã cài đặt nó.

  • Tạo một tensor và in nó.

  • Tính toán torch.log (đầu vào) . Cần đầu vào , tensor, làm tham số đầu vào và trả về tensor mới với các giá trị logarit tự nhiên của các phần tử của đầu vào .

  • In ra tensor với các giá trị logarit tự nhiên của các phần tử của tensor đầu vào ban đầu.

Ví dụ 1

Chương trình Python sau đây chỉ ra cách tính logarit tự nhiên của tensor PyTorch.

# import necessary library
import torch

# Create a tensor
t = torch.Tensor([2.3,3,2.3,4,3.4])

# print the above created tensor
print("Original tensor:\n", t)

# compute the logarithm of elements of the above tensor
log = torch.log(t)

# print the computed logarithm of elements
print("Logarithm of Elements:\n", log)

Đầu ra

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])

Ví dụ 2

Chương trình Python sau đây chỉ ra cách tính logarit tự nhiên của tensor 2D.

# import necessary libraries
import torch

# Create a tensor of random numbers of size 3x4
t = torch.rand(3,4)

# print the above created tensor
print("Original tensor:\n", t)

# compute the logarithm of elements of the above tensor
log = torch.log(t)

# print the computed logarithm of elements
print("Logarithm of Elements:\n", log)

Đầu ra

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]])