这里整理了一下关于Pytorch中torch.linspace
的用法,主要参考了pytorch的官方文档。
linspace
是linear space的缩写,中文含义是线性等分量。
下面是原函数
torch.linspace(start, end, steps, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
功能
在区间[start, end] 上生成均匀的N个点
主要参数
- start(float) - 区间的起始点
- end (float) - 区间的终点
- steps(int)- 在start 和 end间生成的样本数
- out(Tensor, optional) - 结果张量
start可以比end小,也可以比end大。
import torch
torch.linspace(3,10, steps = 5)
# tensor([ 3.0000, 4.7500, 6.5000, 8.2500, 10.0000])
torch.linspace(-10,10,steps=5)
# tensor([-10., -5., 0., 5., 10.])
torch.linspace(-10,10,1)
#tensor([-10.])