mindspore.ops.range

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

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

说明

三个输入必须全为整数或全为浮点数。

参数:
  • start (number) - 区间的起始值。

  • end (number) - 区间的末尾数。

  • step (number) - 值的间隔。

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

返回:

Tensor

支持平台:

GPU CPU

样例:

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