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()