megengine.traced_module.wrap# wrap(func)[源代码]# 调用这个函数将 func 注册为一个内置函数。 这个函数可以在模块级作用域被调用,来将 func 注册为一个内置函数。在 trace 时,内置函数将被转换为 CallFunction Expr. def my_func(x, y): return x + y import megengine.traced_module as tm tm.wrap(my_func) 这个函数也可以用作装饰器: @tm.wrap def my_func(x, y): return x + y 参数: func (Callable) – 当被调用时要插入到计算图中的全局函数。