当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch sin()用法及代码示例


PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。

函数torch.sin()提供对PyTorch中正弦函数的支持。它期望输入为弧度形式,并且输出范围为[-1,1]。输入类型为张量,如果输入包含多个元素,则将按元素计算正弦。

用法: torch.sin(x, out=None)

参数
x:输入张量
name(可选):输出张量

返回类型:与x具有相同类型的张量。

代码1:

# Importing the PyTorch library 
import torch 
  
# A constant tensor of size 6 
a = torch.FloatTensor([1.0, -0.5, 3.4, -2.1, 0.0, -6.5]) 
print(a) 
  
# Applying the sin function and 
# storing the result in 'b' 
b = torch.sin(a) 
print(b)

输出:

 1.0000
-0.5000
 3.4000
-2.1000
 0.0000
-6.5000
[torch.FloatTensor of size 6]


 0.8415
-0.4794
-0.2555
-0.8632
 0.0000
-0.2151
[torch.FloatTensor of size 6]

代码2:可视化

# Importing the PyTorch library 
import torch 
  
# Importing the NumPy library 
import numpy as np 
  
# Importing the matplotlib.pylot function 
import matplotlib.pyplot as plt 
  
# A vector of size 15 with values from -5 to 5 
a = np.linspace(-5, 5, 15) 
  
# Applying the sine function and 
# storing the result in 'b' 
b = torch.sin(torch.FloatTensor(a)) 
  
print(b) 
  
# Plotting 
plt.plot(a, b.numpy(), color = 'red', marker = "o")  
plt.title("torch.sin")  
plt.xlabel("X")  
plt.ylabel("Y")  
  
plt.show()

输出:

 0.9589
 0.9103
 0.4167
-0.2806
-0.8408
-0.9899
-0.6551
 0.0000
 0.6551
 0.9899
 0.8408
 0.2806
-0.4167
-0.9103
-0.9589
[torch.FloatTensor of size 15]



相关用法


注:本文由纯净天空筛选整理自vaibhav29498大神的英文原创作品 Python | PyTorch sin() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。