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