PyTorch是由Facebook开发的开源机器学习库。它用于深度神经网络和自然语言处理。
函数torch.linspace()
返回一维步长张量,该步张量在起点和终点之间等距。
输出张量是尺寸步长的一维。
用法:torch.linspace(start, end, steps=100, out=None)
参数:
start:点集的起始值。
end:点集的最终值
steps:每对相邻点之间的间隙。默认值:100
out(Tensor, optional):输出张量
返回类型:张量
代码1:
# Importing the PyTorch library
import torch
# Applying the linspace function and
# storing the resulting tensor in 't'
a = torch.linspace(3, 10, 5)
print("a = ", a)
b = torch.linspace(start =-10, end = 10, steps = 5)
print("b = ", b)
输出:
a = tensor([ 3.0000, 4.7500, 6.5000, 8.2500, 10.0000]) b = tensor([-10., -5., 0., 5., 10.])
代码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
# Applying the linspace function to get a tensor of size 15 with values from -5 to 5
a = torch.linspace(-5, 5, 15)
print(a)
# Plotting
plt.plot(a.numpy(), np.zeros(a.numpy().shape), color = 'red', marker = "o")
plt.title("torch.linspace")
plt.xlabel("X")
plt.ylabel("Y")
plt.show()
输出:
tensor([-5.0000, -4.2857, -3.5714, -2.8571, -2.1429, -1.4286, -0.7143, 0.0000, 0.7143, 1.4286, 2.1429, 2.8571, 3.5714, 4.2857, 5.0000]) [torch.FloatTensor of size 15]
相关用法
- Python numpy.linspace()用法及代码示例
- Python PyTorch sin()用法及代码示例
- Python PyTorch sinh()用法及代码示例
- Python PyTorch cosh()用法及代码示例
- Python PyTorch tanh()用法及代码示例
- Python PyTorch cos()用法及代码示例
- Python PyTorch tan()用法及代码示例
- Python PyTorch asin()用法及代码示例
- Python PyTorch acos()用法及代码示例
- Python PyTorch atan()用法及代码示例
- Python PyTorch zeros()用法及代码示例
- Python Pytorch ones()用法及代码示例
- Python Pytorch arange()用法及代码示例
- Python Pytorch range()用法及代码示例
- Python Pytorch logspace()用法及代码示例
- Python Pytorch eye()用法及代码示例
- Python Pytorch empty()用法及代码示例
- Python Pytorch full()用法及代码示例
- Python PyTorch floor()用法及代码示例
注:本文由纯净天空筛选整理自sanskar27jain大神的英文原创作品 Python Pytorch linspace() method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。