megengine.traced_module.trace_module

trace_module(mod, *args, **kwargs)[源代码]

追溯 mod 模块并返回对应的 TracedModule.

参数

  • mod (Module) – 这个模块将被转成 TracedModule.

  • args (Tuple[Any]) – 传递给 mod 的 forward 方法的位置参数。

  • kwargs (Dict[str, Any]) – 传递给 mod 的 forward 方法的关键字参数。

返回类型

TracedModule

上一页

TracedModule

下一页

megengine.traced_module.register_as_builtin