torch.nn.modules.module.register_module_full_backward_hook#
- torch.nn.modules.module.register_module_full_backward_hook(hook)[原始碼]#
註冊所有模組通用的後向鉤子。
警告
此函式會向 nn.module 模組新增全域性狀態，僅用於除錯/剖析目的。
使用此函式註冊的鉤子行為與使用
torch.nn.Module.register_full_backward_hook()註冊的鉤子行為相同。有關更多詳細資訊，請參閱其文件。使用此函式註冊的鉤子將在使用
torch.nn.Module.register_full_backward_hook()註冊的鉤子之前呼叫。- 返回
一個控制代碼，可用於透過呼叫
handle.remove()來移除新增的鉤子- 返回型別
torch.utils.hooks.RemovableHandle
