PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。
函数torch.cos()
为PyTorch中的余弦函数提供支持。它期望输入为弧度形式,并且输出范围为[-1,1]。输入类型为张量,如果输入包含多个元素,则将计算按元素的余弦值。
用法: torch.cos(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 cos function and
# storing the result in 'b'
b = torch.cos(a)
print(b)
输出:
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:可视化
# 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 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]
相关用法
- Python PyTorch sin()用法及代码示例
- Python PyTorch tan()用法及代码示例
- Python PyTorch asin()用法及代码示例
- Python PyTorch tanh()用法及代码示例
- Python PyTorch sinh()用法及代码示例
- Python PyTorch cosh()用法及代码示例
- Python PyTorch acos()用法及代码示例
- Python PyTorch atan()用法及代码示例
- Python set()用法及代码示例
- Python os.dup()用法及代码示例
- Python next()用法及代码示例
注:本文由纯净天空筛选整理自vaibhav29498大神的英文原创作品 Python | PyTorch cos() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。