如何在 PyTorch 中计算张量元素的正弦?

pythonpytorchserver side programmingprogramming

要计算张量元素的正弦,我们使用 torch.sin() 方法。它返回一个新张量,其中包含原始输入张量元素的正弦值。它将张量作为输入参数并输出张量。

步骤

  • 导入所需的库。在以下所有 Python 示例中,所需的 Python 库是 torch。确保您已经安装了它。

  • 创建一个张量并打印它。

  • 计算 torch.sin(input)。它以 input(一个张量)作为输入参数,并返回一个包含输入元素正弦值的新张量。

  • 打印包含原始输入张量元素正弦值的张量。

示例 1

# 用于计算张量元素正弦值的 Python 程序
# 导入必要的库
import torch

# 创建张量
T = torch.Tensor([1.3,4.32,4.4,5.3,4.5])
print("Original Tensor T:\n", T)

# 计算上述张量的正弦
sine_T = torch.sin(T)
print("张量元素正弦值T:\n", sine_T)

输出

原始张量 T:
   tensor([1.3000, 4.3200, 4.4000, 5.3000, 4.5000])
张量 T 元素的正弦值:
   tensor([ 0.9636, -0.9240, -0.9516, -0.8323, -0.9775])

示例 2

# Python 程序计算张量元素的正弦
# 导入必要的库
import torch

# 创建大小为 3x5 的 2D 张量
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("原始张量 T:\n", T)

# 计算上述张量的正弦
sine_T = torch.sin(T)
print("张量 T 元素的正弦值:\n", sine_T)

输出

原始张量 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]])
张量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]])

相关文章