torch.nn.modules.module.register_module_full_backward_pre_hook¶
- torch.nn.modules.module.register_module_full_backward_pre_hook(hook)[source][source]¶
註冊一個適用於所有模組的後向前置鉤子。
警告
這會向 nn.module 模組新增全域性狀態,並且僅用於除錯/效能分析目的。
使用此函式註冊的鉤子與使用
torch.nn.Module.register_full_backward_pre_hook()註冊的鉤子行為方式相同。有關更多詳細資訊,請參閱其文件。使用此函式註冊的鉤子將優先於使用
torch.nn.Module.register_full_backward_pre_hook()註冊的鉤子被呼叫。- 返回
一個控制代碼,可以透過呼叫
handle.remove()來移除新增的鉤子- 返回型別
torch.utils.hooks.RemovableHandle