megengine.traced_module.TracedModule.register_forward_pre_hook

TracedModule.register_forward_pre_hook(hook)

Registers a hook to handle forward inputs. hook should be a function.

参数

hook (Callable) – a function that receive module and inputs, then return a modified inputs or None.

返回类型

HookHandler

返回

a handler with remove interface to delete the hook.

上一页

megengine.traced_module.TracedModule.register_forward_hook

下一页

megengine.traced_module.TracedModule.replace_param