torch.linspace()

PyTorch中的torch.linspace

 linspace是linear space的缩写，中文含义是线性等分向量

原函数

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

参数

start： 开始值
end：结束值
steps：分割的点数，默认为100

示例代码

import torch
print(torch.linspace(1, 10, 5))

#输出结果：
	tensor([ 1.0000,  3.2500,  5.5000,  7.7500, 10.0000])