torch.linspace

函数原型
torch.linspace(start, end, steps=100, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

功能
返回一个一维的张量

参数意义（一般只用前三个参数）
start (float) – 起始点
end (float) – 结束点
steps (int) – 共有多少点（在起点和中点间，包括起点和终点）. 默认: 100.
out (Tensor, optional) – 输出的张量
dtype (torch.dtype, optional) –想要返回的数据类型(torch.dtype). 如果是None，就是用全局的默认数据类型（通过torch.set_default_tensor_type()设置）
layout (torch.layout, optional) –返回张量的期望布局. 默认: torch.strided.
device (torch.device, optional) – CPU或GPU
requires_grad (bool, optional) – 记录返回张量的操作. Default: False.（但在机器学习中长设为True）
最最常用的参数 start end steps

例子

>>> import torch
>>> torch.linspace(3,5,10)
tensor([3.0000, 3.2222, 3.4444, 3.6667, 3.8889, 4.1111, 4.3333, 4.5556, 4.7778,
        5.0000])
>>> torch.linspace(start = 3,end = 5,steps =10)
tensor([3.0000, 3.2222, 3.4444, 3.6667, 3.8889, 4.1111, 4.3333, 4.5556, 4.7778,
        5.0000])