mindspore.ops.range

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

返回从 start 开始,步长为 step ,且不超过 end (不包括 end )的序列。

三个输入的数据类型必须相同。函数返回的Tensor的数据类型与输入数据类型保持一致。

参数:
  • start (Tensor) - 标量Tensor,序列中的第一个数字。数据类型必须为int32,int64,float32或者float64。

  • end (Tensor) - 标量Tensor,序列中的数值上限,不包括其本身。数据类型必须为int32,int64,float32或者float64。

  • step (Tensor) - 标量Tensor,表述序列中数值的步长。数据类型必须为int32,int64,float32或者float64。

返回:

一维Tensor,数据类型与输入数据类型一致。

异常:
  • TypeError - startendstep 不是标量Tensor。

  • TypeError - startendstep 的数据类型不一致。

  • TypeError - startendstep 的数据类型不支持。

  • ValueError - step 等于0。

  • ValueError - start 小于等于 endstep 小于0。

  • ValueError - start 大于等于 endstep 大于0。

支持平台:

GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> from mindspore import dtype as mstype
>>> start = Tensor(0, mstype.int32)
>>> end = Tensor(10, mstype.int32)
>>> step = Tensor(4, mstype.int32)
>>> output = ops.range(start, end, step)
>>> print(output)
[0 4 8]