注意,torch.linsapce 第三个参数的含义与 range函数第三个参数的含义不同。
例子1
import torch
print(torch.linspace(-5,5,11))
注意,第三个参数的含义是
被采样的元素的数目
在这里的例子中是 11 不是 10!
输出:
tensor([-5., -4., -3., -2., -1., 0., 1., 2., 3., 4., 5.])
例子2
import torch
print(torch.linspace(-5,5,10))
当取10 时
输出
tensor([-5.0000, -3.8889, -2.7778, -1.6667, -0.5556, 0.5556, 1.6667, 2.7778,
3.8889, 5.0000])