megengine.functional.arange¶
- arange(start=0, stop=None, step=1, *, dtype='float32', device=None)[源代码]¶
返回半开区间 [start, stop) 内均匀间隔的值组成的1维Tensor。
注解
step 不是整数且浮点舍入误差影响了输出Tensor长度的情况下，本函数无法保证区间中不包含 stop 值。
- 参数
start (Number) – 如果
stop被指定了， 该值表示区间的开始(区间包含该值)；否则该值表示区间的结束(区间不包含该值)。如果stop没有被指定， 默认的区间开始值是0。stop (Number) – 区间的终点
step (Number) – 两个相邻元素之间的距离 (
out[i+1] - out[i]). 不能为0但可以为负数，如果为负数且stop>=start, 则结果是一个空张量
- 关键字参数
dtype (
Tensor.dtype, optional) – 输出向量的数据类型device (
Tensor.device, optional) – 被创建的向量所处的设备
参见
- 返回类型
- 返回
由均匀间隔的值组成的1维tensor。在
stop - start和step符号相同时输出 tensor 的长度是ceil((stop-start)/step)，其它情况下是0。
实际案例
>>> F.arange(5) Tensor([0. 1. 2. 3. 4.], device=xpux:0) >>> F.arange(1, 4) Tensor([1. 2. 3.], device=xpux:0)