torch.nn.modules.module.register_module_module_registration_hook

torch.nn.modules.module.register_module_module_registration_hook(hook)

注册所有模块通用的模块注册钩子。

警告

这会将全局状态添加到 nn.Module 模块

每次调用 register_module() 时都会调用该钩子。它应具有以下签名

hook(module, name, submodule) -> None or new submodule

钩子可以修改输入或在钩子中返回单个修改后的值。

返回

一个句柄，可用于通过调用 handle.remove() 来删除添加的钩子

返回类型

torch.utils.hooks.RemovableHandle