mindspore.ops.range

查看源文件
mindspore.ops.range(start, end, step, maxlen=1000000)[源代码]

返回一个在 [ start, end ) 区间内,步长为 step 的tensor。

说明

  • 三个输入的数据类型必须全为整数或全为浮点数。

  • 当输入为tensor时,tensor中仅可包含一个元素,数据类型为数值型。

参数:
  • start (Union[Number, Tensor]) - 区间的起始值。

  • end (Union[Number, Tensor]) - 区间的末尾数。

  • step (Union[Number, Tensor]) - 值的间隔。

  • maxlen (int,可选) - 该算子将会被分配能够存储 maxlen 个数据的内存。该参数是可选的,必须为正数,默认 1000000 。如果输出的数量超过 maxlen ,将会引起运行时错误。

返回:

Tensor

支持平台:

GPU CPU

样例:

>>> import mindspore
>>> mindspore.ops.range(0, 6, 2)
Tensor(shape=[3], dtype=Int64, value= [0, 2, 4])