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

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

函数torch.atan()为PyTorch中的反正切函数提供支持。它以弧度形式给出输出。输入类型为张量，如果输入包含多个元素，则将计算按元素的反正切。

用法: torch.atan(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, 0.2, 0.0, -2]) 
print(a) 
  
# Applying the inverse tan function and 
# storing the result in 'b' 
b = torch.atan(a) 
print(b)

输出：

tensor([ 1.0000, -0.5000,  3.4000,  0.2000,  0.0000, -2.0000])
tensor([ 0.7854, -0.4636,  1.2847,  0.1974,  0.0000, -1.1071])

代码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 inverse tangent function and 
# storing the result in 'b' 
b = torch.atan(torch.FloatTensor(a)) 
  
print(b) 
  
# Plotting 
plt.plot(a, b.numpy(), color = 'red', marker = "o")  
plt.title("torch.atan")  
plt.xlabel("X")  
plt.ylabel("Y")  
  
plt.show()

输出：

tensor([-1.3734, -1.3416, -1.2978, -1.2341, -1.1342, -0.9601, -0.6202,  0.0000,
         0.6202,  0.9601,  1.1342,  1.2341,  1.2978,  1.3416,  1.3734])

相关用法

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