Python| PyTorch cos() 方法
PyTorch 是 Facebook 开发的开源机器学习库。它用于深度神经网络和自然语言处理目的。
函数torch.cos() 为 PyTorch 中的余弦函数提供支持。它期望以弧度形式输入,输出在 [-1, 1] 范围内。输入类型是张量,如果输入包含多个元素,则计算元素余弦。
Syntax: torch.cos(x, out=None)
Parameters:
x: Input tensor
name (optional): Output tensor
Return type: A tensor with the same type as that of x.
代码#1:
Python3
# 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 cos function and
# storing the result in 'b'
b = torch.cos(a)
print(b)
Python3
# Importing the PyTorch library
import torch
# Importing the NumPy library
import numpy as np
# Importing the matplotlib.pyplot 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 cosine function and
# storing the result in 'b'
b = torch.cos(torch.FloatTensor(a))
print(b)
# Plotting
plt.plot(a, b.numpy(), color = 'red', marker = "o")
plt.title("torch.cos")
plt.xlabel("X")
plt.ylabel("Y")
plt.show()
输出:
1.0000
-0.5000
3.4000
-2.1000
0.0000
-6.5000
[torch.FloatTensor of size 6]
0.5403
0.8776
-0.9668
-0.5048
1.0000
0.9766
[torch.FloatTensor of size 6]
代码 #2:可视化
Python3
# Importing the PyTorch library
import torch
# Importing the NumPy library
import numpy as np
# Importing the matplotlib.pyplot 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 cosine function and
# storing the result in 'b'
b = torch.cos(torch.FloatTensor(a))
print(b)
# Plotting
plt.plot(a, b.numpy(), color = 'red', marker = "o")
plt.title("torch.cos")
plt.xlabel("X")
plt.ylabel("Y")
plt.show()
输出:
0.2837
-0.4138
-0.9090
-0.9598
-0.5414
0.1417
0.7556
1.0000
0.7556
0.1417
-0.5414
-0.9598
-0.9090
-0.4138
0.2837
[torch.FloatTensor of size 15]