megengine.traced_module.TracedModule.apply

TracedModule.apply(fn)

Applies function fn to all the modules within this module, including itself.

参数

fn (Callable[[Module], Any]) – the function to be applied on modules.

返回类型

None

上一页

megengine.traced_module.TracedModule.graph

下一页

megengine.traced_module.TracedModule.buffers