forked from mindspore-Ecosystem/mindspore
101 lines
2.9 KiB
ReStructuredText
101 lines
2.9 KiB
ReStructuredText
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是否会更新参数的值。
|