pytorch 中 torch.linspace()在某区间生成均匀的N个点

函数详解：

torch.linspace(start, end, steps, out)

• 功能：在区间【start，end】上生成均匀的N个点。
• 参数：
  • start（float）：区间的起点；
  • end（float）：区间的终点；
  • steps（int）：生成的点数；
  • out（Tensor，optional）：结果张量。
• 注意：start 可以比 end 小，也可以比 end 大，也可以和 end 相等。

示例：

import torch

a = torch.linspace(0, 1, 5)
print(a)  ## tensor([0.0000, 0.2500, 0.5000, 0.7500, 1.0000])

a = torch.linspace(1, 0, 5)
print(a)  ## tensor([1.0000, 0.7500, 0.5000, 0.2500, 0.0000])

a = torch.linspace(1, 1, 5)
print(a)  ## tensor([1., 1., 1., 1., 1.])