torch.nn.modules.module.register_module_full_backward_hook¶
- torch.nn.modules.module.register_module_full_backward_hook(hook)[source][source]¶
註冊一個適用於所有模組的後向鉤子。
警告
這會向 nn.module 模組新增全域性狀態，僅用於除錯/效能分析目的。
使用此函式註冊的鉤子與使用
torch.nn.Module.register_full_backward_hook()註冊的鉤子行為相同。詳情請參考其文件。使用此函式註冊的鉤子將在使用
torch.nn.Module.register_full_backward_hook()註冊的鉤子之前呼叫。- 返回值
一個控制代碼，透過呼叫
handle.remove()可用於移除新增的鉤子- 返回型別
torch.utils.hooks.RemovableHandle
