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)
,就生成整型啦。
如果你喜欢的话...
如果读完我写的笔记有疑问或者想法,欢迎留下您的评论,我们一起交流、共同讨论、相互学习。如果这篇笔记让您有收获,愿您不吝打赏,您的鼓励是对我最大的肯定,也督促我记录更多质量更好的笔记。
![打赏码](https://www.cnblogs.com/images/cnblogs_com/datasnail/1294095/o_QQ%E6%88%AA%E5%9B%BE20180905125739.png)