老饼讲解-深度学习 机器学习 神经网络 深度学习
pytorch入门
1.pytorch学前准备
2.pytorch的基础操作
3.pytorch的梯度计算
4.pytorch的数据
5.pytorch-小试牛刀

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

作者 : 老饼 日期 : 2023-07-28 10:49:54 更新 : 2024-01-19 08:11:10
本站原创文章,转载请说明来自《老饼讲解-深度学习》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 




联系老饼