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