arange

arange

paddle.fluid.layers.arange(start, end, step=1, dtype=None, name=None)[源代码]

该API根据step均匀分隔给定数值区间[start, end)，并返回该分隔结果。

参数

start （float32 | float64 | int32 | int64 | Variable） - 区间起点，且区间包括此值, 当类型是Variable时，是shape为 [1] 的1-D Tensor。
end （float32 | float64 | int32 | int64 | Variable） - 区间终点，通常区间不包括此值。但当step不是整数，且浮点数取整会影响输出的长度时例外。
step （float32 | float64 | int32 | int64 | Variable） - 均匀分割的步长。
dtype （str | core.VarDesc.VarType） - 输出Tensor的数据类型，可为 ‘float32’, ‘float64’, ‘int32’, ‘int64’ 。

均匀分割给定数值区间后得到的1-D Tensor, 数据类型为输入 dtype 。

返回类型

Variable

代码示例

import paddle
import paddle.fluid as fluid
with fluid.dygraph.guard():
             x = fluid.layers.arange(0, 6, 2)
             # x: [0, 2, 4]
             # x dtype: float32