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