Python | PyTorch sin() 方法

简介

sin() 方法是 PyTorch 中的数学函数之一，用于计算一个tensor中每个元素的正弦值。它适用于浮点数和整数类型的tensor，但数据类型必须是float或double。

此方法将返回一个新的tensor，其形状与输入tensor相同。

语法

torch.sin(input, out=None) -> Tensor

参数说明:

• input (Tensor) - 输入的tensor
• out (Tensor, 可选) - 指定输出结果的tensor

如果out未提供，则结果将存储在新创建的tensor中。

示例

import torch

input = torch.tensor([0, 30, 45, 60, 90])

# 计算tensor中每个元素的正弦值
output = torch.sin(input)

print("input tensor: ", input)
print("output tensor: ", output)

输出结果如下：

input tensor: tensor([ 0, 30, 45, 60, 90])
output tensor: tensor([ 0.0000, -0.9880,  0.8509, -0.3048,  0.8930])

更多示例

下面我们将使用 PyTorch 中的 sin() 方法生成正弦曲线：

import torch
import matplotlib.pyplot as plt

# 生成从0到360的一维tensor
x = torch.linspace(0, 2 * 3.14, 300)

# 计算sin函数的值
y = torch.sin(x)

# 将tensor转换为numpy array
x = x.numpy()
y = y.numpy()

# 绘制图像
plt.plot(x, y)
plt.show()

输出结果如下：

总结

sin() 方法是 PyTorch 中非常有用的数学函数之一。它用于计算tensor中每个元素的正弦值。通过示例，我们学习了如何使用该函数在 PyTorch 中生成正弦曲线。