torch.nn.modules.module.register_module_parameter_registration_hook

torch.nn.modules.module.register_module_parameter_registration_hook(hook)[源代码]

为所有模块注册一个通用的参数注册钩子。

警告

这向nn.Module模块添加了全局状态。

每次调用 register_parameter() 时都会触发此钩子。它应具有以下签名：

hook(module, name, param) -> None or new parameter

该钩子可以修改输入，并返回一个修改后的值。