函数定义:
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])