mindspore/docs/api/api_python/ops/mindspore.ops.func_arange.rst

24 lines
1.4 KiB
ReStructuredText
Raw Normal View History

2022-11-18 15:13:10 +08:00
mindspore.ops.arange
=====================
.. py:function:: mindspore.ops.arange(start=0, end=None, step=1, *, dtype=None)
返回从 `start` 开始,步长为 `step` ,且不超过 `end` (不包括 `end` )的序列。
参数:
2022-11-21 11:08:17 +08:00
- **start** (Union[float, int, Tensor], 可选) - 序列中的第一个数字。如果为Tensor则shape必须为()。默认值0。
- **end** (Union[float, int, Tensor], 可选) - 序列的上限或下限不包含在序列中。如果为Tensor则shape必须为()。默认值None。如果为None则默认为 `start` 的值同时将0作为范围起始值。
- **step** (Union[float, int, Tensor], 可选) - 表述序列中数值的步长。如果为Tensor则shape必须为()。默认值1。
关键字参数:
2023-02-09 20:38:48 +08:00
- **dtype** (mindspore.dtype, 可选) - 返回Tensor的所需数据类型。默认值None。如果未指定或者为None将会被推断为 `start``end``step` 参数中精度最高的类型。
2022-11-18 15:13:10 +08:00
返回:
一维Tensor数据类型与输入数据类型一致。
异常:
2022-11-21 11:08:17 +08:00
- **TypeError** - `start` `end` `step` 既不是int也不是float也不是在支持类型中的TensorScalar(shape为()的特殊Tensor)。
2022-11-18 15:13:10 +08:00
- **ValueError** - `step` 等于0。
2022-11-21 11:08:17 +08:00
- **ValueError** - `start` 小于等于 `end` ,且 `step` 小于0。
- **ValueError** - `start` 大于等于 `end` ,且 `step` 大于0。