48 lines
2.2 KiB
ReStructuredText
48 lines
2.2 KiB
ReStructuredText
mindspore.ops.Scan
|
||
====================
|
||
|
||
.. py:class:: mindspore.ops.Scan
|
||
|
||
将一个函数循环作用于一个数组，且对当前元素的处理依赖上一个元素的执行结果。
|
||
Scan算子的执行逻辑可以近似表示为如下代码:
|
||
|
||
.. code-block:: python
|
||
|
||
def Scan(loop_func, init, xs, length=None):
|
||
if xs is None:
|
||
xs = [None] * length
|
||
carry = init
|
||
ys = []
|
||
for x in xs:
|
||
carry, y = loop_func(carry, x)
|
||
ys.append(y)
|
||
return carry, ys
|
||
|
||
当前Scan算子存在以下语法限制:
|
||
|
||
- 暂不支持 `loop_func` 为副作用函数，如：对Parameter、全局变量的修改等操作。
|
||
- 暂不支持 `loop_func` 的返回值的第一个元素与初始值 `init` 的类型或形状不同。
|
||
|
||
.. warning::
|
||
这是一个实验性API，后续可能修改或删除。
|
||
|
||
输入：
|
||
- **loop_func** (Function) - 循环体函数。
|
||
- **init** (Union[Tensor, number, str, bool, list, tuple, dict]) - 循环的初始值。
|
||
- **xs** (Union[tuple, list, None]) - 用于执行循环扫描的数组。
|
||
- **length** (Union[int, None], 可选) - 数组xs的长度，默认值： ``None`` 。
|
||
- **unroll** (bool, 可选) - 是否在编译阶段进行循环展开，默认值： ``True`` 。
|
||
|
||
输出：
|
||
Tuple(Union[Tensor, number, str, bool, list, tuple, dict], list)，
|
||
由两个元素组成的tuple，第一个元素为循环的最终结果，和 `init` 参数保持一样的类型和形状；
|
||
第二个元素是一个列表，包含每次循环的执行结果。
|
||
|
||
异常：
|
||
- **TypeError** - `loop_func` 不是一个函数。
|
||
- **TypeError** - `xs` 不是一个tuple、一个list或者None。
|
||
- **TypeError** - `length` 不是一个整数或者None。
|
||
- **TypeError** - `unroll` 不是一个布尔值。
|
||
- **ValueError** - `loop_func` 不能接受 `init` 以及 `xs` 的元素作为参数。
|
||
- **ValueError** - `loop_func` 的返回值不是一个包含两个元素的tuple，或者tuple的第一个元素与 `init` 的类型或形状不同。
|