【例子】pytorch-tensor生成等差序列

本站原创文章，转载请说明来自《老饼讲解-深度学习》www.bbbdata.com

等差序列是编程中经常要用到的一种序列变量，例如用于循环时的索引等等

本文讲解如何在pytorch中生成等差序列的tensor,并展示对应的运行结果

一、pytorch生成等差序列tensor   

本节讲解在pytorch中两种生成等差序列tensor的方法

1.按步长生成等差序列

在pytorch中要按步长生成等差序列，可以使用arange

例如:从1开始，逐步增加0.5，直到4结束（不包括4）

x = torch.arange(1, 4,0.5)

运行结果如下：

tensor([1.0000, 1.5000, 2.0000, 2.5000, 3.0000, 3.5000])

2. 按序列长度生成等差序列

在pytorch中要生成固定序列长度的等差序列，可以使用linspace

例如从-5到5之间等间隔生成20个数，可以如下实现

#从-5到5之间等间隔生成20个数
x = torch.linspace(-5,5,20)

运行结果如下：

tensor([-5.0000, -4.4737, -3.9474, -3.4211, -2.8947, -2.3684, -1.8421, -1.3158,
        -0.7895, -0.2632,  0.2632,  0.7895,  1.3158,  1.8421,  2.3684,  2.8947,
         3.4211,  3.9474,  4.4737,  5.0000])

好了，以上就是如何在pytorch中生成等差序列的tensor了~

 End