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

該鉤子可以修改輸入或在鉤子中返回單個修改後的值。

返回

一個控制代碼，可以透過呼叫 handle.remove() 來移除新增的鉤子

返回型別

torch.utils.hooks.RemovableHandle