torch.linspace(start, end, steps, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor
创建一个大小 steps
为一维张量,其值从 start
到 (包括 ) end
均匀分布。
区间起点start = a 区间终点end = b 元素数量step = n
举例说明:
>>> torch.linspace(-1,1,3)
tensor([-1., 0., 1.])
>>> torch.linspace(-10,10,10)
tensor([-10.0000, -7.7778, -5.5556, -3.3333, -1.1111, 1.1111, 3.3333,
5.5556, 7.7778, 10.0000])
!!!!!!闭区间 左闭右闭
所以np.linspace(0,2,3) 就是[0 1 2]
从-1到1,等距分布,共3个元素。
从-10到10,等距分布,共10个元素。