torch.nn.modules.module.register_module_parameter_registration_hook#
- torch.nn.modules.module.register_module_parameter_registration_hook(hook)[源代码]#
为所有模块注册一个通用的参数注册钩子。
警告
This adds global state to the nn.Module module
每次调用
register_parameter()时都会调用该钩子。它应该具有以下签名：hook(module, name, param) -> None or new parameter
The hook can modify the input or return a single modified value in the hook.
- 返回
一个句柄，可用于通过调用
handle.remove()来移除添加的钩子- 返回类型
torch.utils.hooks.RemovableHandle
