megengine.traced_module#

>>> import megengine.traced_module as tm

TracedModule

TracedModule 是通过追溯(tracing)普通模块创建的 Module.

trace_module

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

register_as_builtin

将类 mod_clsModule 的子类)注册为内置模块。

wrap

调用这个函数将 func 注册成内置函数。

节点#

ModuleNode

ModuleNode 表示 Module 对象。

TensorNode

TensorNode 表示 Tensor 对象。

Expr#

Input

一个用来标记图的输入的假 Expr.

GetAttr

Getattr 表示从 Module 的层次结构中获取属性。

CallMethod

CallMethod 表示对 Module 的 __call__ 方法或 Tensor 的方法的调用。

CallFunction

CallFunction 表示对内置函数的调用。

Constant

Constant 表示非 Module属性的 TensorModule.

Apply

Apply 表示对 apply 的调用。

InternalGraph#

InternalGraph

InternalGraph 是 TracedModule 中使用的主要数据结构。