torch.nn.modules.module.register_module_backward_hook#
- torch.nn.modules.module.register_module_backward_hook(hook)[原始碼]#
註冊所有模組通用的後向鉤子。
此函式已棄用,建議使用
torch.nn.modules.module.register_module_full_backward_hook(),並且此函式在未來版本中的行為將發生變化。- 返回
一個控制代碼,可用於透過呼叫
handle.remove()來移除新增的鉤子- 返回型別
torch.utils.hooks.RemovableHandle