mindspore.ops.constexpr

mindspore.ops.constexpr(fn=None, get_instance=True, name=None, reuse_result=True)[源代码]

创建PrimiveWithInfer算子,用于在编译时推断值。可以用它定义函数,从而使用构造函数中的常量计算出常量值。

参数:

  • fn (function) - fn 用作输出算子的infer_value。默认值:None。

  • get_instance (bool) - 如果为True,返回算子的实例,否则返回算子的类。默认值:True。

  • name (str) - 定义算子的名称。如果 name 为None,则使用函数名称作为算子名称。默认值:None。

  • reuse_result (bool) - 如果为True,仅实际执行算子一次,后续调用会直接返回结果。否则每次都实际执行算子来获取结果。默认值:True。

样例:

>>> from mindspore.ops import constexpr
>>> a = (1, 2)
>>> # make an operator to calculate tuple len
>>> @constexpr
... def tuple_len(x):
...     return len(x)
...
>>> print(tuple_len(a))
2
>>> # make an operator class to calculate tuple len
>>> @constexpr(get_instance=False, name="TupleLen")
... def tuple_len_class(x):
...     return len(x)
...
>>> print(tuple_len_class()(a))
2