mindspore.ops.prim_attr_register

mindspore.ops.prim_attr_register(fn)[源代码]

Primitive属性的注册器。

注册装饰器,其中装饰器用于内置算子的Primitive的’__init__’函数。该函数将添加’__init__’的所有参数作为算子属性,并且初始化Primitive的名称。

参数:

  • fn (function) - Primitive的__init__函数。

返回:

函数,原始函数。

样例:

>>> from mindspore.ops.primitive import prim_attr_register, PrimitiveWithCheck
>>> class MatMul(PrimitiveWithCheck):
...     @prim_attr_register
...     def __init__(self, transpose_a=False, transpose_b=False):
...         self.init_prim_io_names(inputs=['x1', 'x2'], outputs=['output'])
...
>>> # init a Primitive obj
>>> matmul = MatMul()