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

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

Returns

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

Return type

torch.utils.hooks.RemovableHandle