torch.nn.modules.module.register_module_parameter_registration_hook

torch.nn.modules.module.register_module_parameter_registration_hook(hook)

註冊一個所有模組通用的參數註冊 hook。

警告

這會將全域狀態新增到 nn.Module 模組

每次呼叫 register_parameter() 時，都會呼叫此 hook。它應具有以下簽名

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

hook 可以修改輸入，或在 hook 中傳回單個修改過的值。

回傳

可用於透過呼叫 handle.remove() 移除新增 hook 的 handle

回傳類型

torch.utils.hooks.RemovableHandle