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

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

Để tính toán sin của các phần tử của một tensor, chúng tôi sử dụng torch.sin () phương pháp. Nó trả về một tensor mới với các giá trị sin 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.sin (đầu vào) . Cần đầu vào , một tensor làm tham số đầu vào và trả về một tensor mới với các giá trị sin của các phần tử của đầu vào.

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

Ví dụ 1

# Python program to compute sine of the elements of a tensor
# import necessary library
import torch

# create a tensor
T = torch.Tensor([1.3,4.32,4.4,5.3,4.5])
print("Original Tensor T:\n", T)

# Compute the sine of above tensor
sine_T = torch.sin(T)
print("Sine value of elements of tensor T:\n", sine_T)

Đầu ra

Original Tensor T:
   tensor([1.3000, 4.3200, 4.4000, 5.3000, 4.5000])
Sine value of elements of tensor T:
   tensor([ 0.9636, -0.9240, -0.9516, -0.8323, -0.9775])

Ví dụ 2

# Python program to compute sine of the elements of a tensor
# import necessary library
import torch

# Create a 2D tensor of size 3x5
T = torch.Tensor([[1.3,4.32,4.4,5.3,4.5],
                  [0.2,0.3,0.5,0.7,0.9],
                  [1.1,1.2,2.3,3.1,4.9]])
print("Original Tensor T:\n", T)

# Compute the sine of above tensor
sine_T = torch.sin(T)
print("Sine value of elements of tensor T:\n", sine_T)

Đầu ra

Original Tensor T:
tensor([[1.3000, 4.3200, 4.4000, 5.3000, 4.5000],
         [0.2000, 0.3000, 0.5000, 0.7000, 0.9000],
         [1.1000, 1.2000, 2.3000, 3.1000, 4.9000]])
Sine value of elements of tensor T:
tensor([[ 0.9636, -0.9240, -0.9516, -0.8323, -0.9775],
         [ 0.1987, 0.2955, 0.4794, 0.6442, 0.7833],
         [ 0.8912, 0.9320, 0.7457, 0.0416, -0.9825]])