PyTorch中的torch.linspace
代码语言:javascript复制 linspace是linear space的缩写,中文含义是线性等分向量
原函数
代码语言:javascript复制torch.linspace(start, end, steps=100, out=None, dtype=None,layout=torch.strided, device=None, requires_grad=False)
参数
代码语言:javascript复制start: 开始值
end:结束值
steps:分割的点数,默认为100
示例代码
代码语言:javascript复制import torch
print(torch.linspace(1, 10, 5))
#输出结果:
tensor([ 1.0000, 3.2500, 5.5000, 7.7500, 10.0000])