mindspore2022/docs/api/api_python/ops/mindspore.ops.Primitive.rst

mindspore.ops.Primitive
=======================

.. py:class:: mindspore.ops.Primitive(name)

    Primitive是Python中算子原语的基类。

    **参数：**

    - **name** (str) - 当前Primitive的名称。

    .. py:method:: add_prim_attr(name, value)

        添加Primitive的属性。

        **参数：**

        - **name** (str) - 属性名称。
        - **value** (Any) - 属性值。

    .. py:method:: check_elim(*args)

        检查是否可以消除此Primitive。有需要的子类可以重写该方法。

        **参数：**

        - **args** (Primitive参数的类型) - 与当前Primitive的参数相同。

        **返回：**

        由两个元素组成的元组。第一个元素是指是否能在编译阶段计算Primitive，第二个元素是计算结果。

    .. py:method:: del_prim_attr(name)

        删除Primitive的属性。

        **参数：**

        - **name** (str) - 属性名称。

    .. py:method:: init_prim_io_names(inputs, outputs)

        初始化Tensor或属性的输入输出的名称。

        **参数：**

        - **inputs** (list[str]) - 输入名称的列表。
        - **outputs** (list[str]) - 输出名称的列表。

    .. py:method:: recompute(mode=True)

        设置Primitive的重计算属性。

        如果有一个被设置了重计算属性的Primitive，并且其结果在计算导数的时候被使用，那么不会保存该Primitive在前向网络中的中间计算结果，而是在自动微分的时候重新进行计算。

        .. note::
            - 如果计算涉及随机化或全局变量，则暂无法保证等效性。
            - 在PyNative模式下不支持。

        **参数：**

        - **mode** (bool) - Primitive是否设置了重计算。默认值：True。

    .. py:method:: set_prim_instance_name(instance_name)

        设置Primitive算子的实例的名称。

        .. note::
            当用户定义Primitive算子时，默认调用它。

        **参数：**

        - **instance_name** (str) - 用户设置的Primitive算子的实例的名称。

    .. py:method:: set_stage(stage)

        将stage的ID添加到Primitive属性中。

        .. note::
            仅在半自动并行模式下有效。在其他并行模式下，请将其设置为0。

        **参数：**

        - **stage** (int) - 当前stage的ID。

    .. py:method:: shard(in_strategy=None, out_strategy=None)

        将切分策略添加到Primitive属性中。

        .. note::
            仅在半自动并行或自动并行模式下有效。在其他并行模式中，将忽略此处设置的策略。

        **参数：**

        - **in_strategy** (tuple) - 描述算子输入的切分策略。默认值：None。
        - **out_strategy** (tuple) - 描述算子输出的切分策略，仅针对某些算子，如MatMul。默认值：None。

    .. py:method:: update_parameter()

        判断此Primitive是否会更新参数的值。