模型开发接口
推理部署接口
API 相关说明
>>> import megengine.traced_module as tm
TracedModule
TracedModule 是通过追溯(tracing)普通模块创建的 Module.
trace_module
追溯 mod 模块并返回对应的 TracedModule.
mod
register_as_builtin
将类 mod_cls ( Module 的子类)注册为内置模块。
mod_cls
Module
wrap
调用这个函数将 func 注册成内置函数。
func
ModuleNode
ModuleNode 表示 Module 对象。
TensorNode
TensorNode 表示 Tensor 对象。
Input
一个用来标记图的输入的假 Expr.
GetAttr
Getattr 表示从 Module 的层次结构中获取属性。
Getattr
CallMethod
CallMethod 表示对 Module 的 __call__ 方法或 Tensor 的方法的调用。
__call__
Tensor
CallFunction
CallFunction 表示对内置函数的调用。
Constant
Constant 表示非 Module属性的 Tensor 或 Module.
Apply
Apply 表示对 apply 的调用。
apply
InternalGraph
InternalGraph 是 TracedModule 中使用的主要数据结构。
上一页
megengine.quantization.fake_quant_tensor
下一页