mindspore.ops.arange

mindspore.ops.arange(start=0, stop=None, step=1, rtype=None)

根据给定的范围返回指定均匀间隔的数据。

参数：

• start (Union[int, float]) - 指定范围的起始值，范围包含该值。类型为int或float。

• stop (Union[int, float]) - 指定范围的结束值，范围不包含该值。类型为int或float。

• step (Union[int, float]) - 指定取值的间隔。类型为int或float。

• rtype (Union[mindspore.dtype, str]) - 指定返回数据的类型，如果不指定，则会根据 start 、 stop 、 step 的值推断类型。

返回：

Tensor，值是均匀间隔的数据，类型为给定或推断的结果。

异常：

• TypeError - start 、 stop 、 step 的类型不是int或float。

• ValueError - start 的值大于等于 stop 。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.ops as ops
>>> print(ops.arange(0, 5, 1))
[0 1 2 3 4]
>>> print(ops.arange(3))
[0 1 2]
>>> print(ops.arange(start=0, stop=3))
[0 1 2]
>>> print(ops.arange(0, stop=3, step=0.5))
[0.  0.5 1.  1.5 2.  2.5]