torch.nn.modules.module.register_module_parameter_registration_hook#
- torch.nn.modules.module.register_module_parameter_registration_hook(hook)[source]#
注册一个参数注册钩子，适用于所有模块。
警告
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
