PyTorch：torch.linspace

函数定义：

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

作用：返回从起始值start(含)到结束值end(含)之间steps个等步距的值的一维张量，步距为(end+1-start)/steps。

有点拗口，直接用例子分析吧：

import torch

a = torch.linspace(start=0, end=9, steps=10)  # torch.linspace(0, 9, 10) 取10个点，步距为1
b = torch.linspace(start=0, end=9, steps=5 )  # torch.linspace(0, 9, 5)  取5个点 ，步距为2
print('a:\n', a)
print('b:\n', b)

'''   运行结果   '''
a:
 tensor([0., 1., 2., 3., 4., 5., 6., 7., 8., 9.])
b:
 tensor([0.0000, 2.2500, 4.5000, 6.7500, 9.0000])