pytorch中的linspace函数

在pytorch中，我们经常会生成一系列的数据，例如我要生成从2,3,4,5,6,7,8,9,10,11十个数字，自己写出来有点傻，所以自动生成是一个好办法。
直接看代码，非常简单：

import torch
torch.linspace(2,11,10,dtype=type(1.0))
‘‘‘
===output===
tensor([ 2.,  3.,  4.,  5.,  6.,  7.,  8.,  9., 10., 11.], dtype=torch.float64)
‘‘‘

这里linspace函数里面参数代表(start, stop, num, dtype)，即（开始的数，结束的数，总共要多少个数，数的类型）。我这里用了type(1.0)，所以最后就生成了是float类型。你也可以把他们换成type(1)，就生成整型啦。

pytorch中的linspace函数

原文：https://www.cnblogs.com/datasnail/p/13086936.html