torch.nn.modules.module.register_module_backward_hook¶
- torch.nn.modules.module.register_module_backward_hook(hook)[源代码][源代码]¶
注册一个适用于所有模块的 backward 钩子。
此函数已弃用,请优先使用
torch.nn.modules.module.register_module_full_backward_hook()
,此函数的行为将在未来版本中更改。- 返回
一个句柄,可以通过调用
handle.remove()
来移除添加的钩子- 返回类型
torch.utils.hooks.RemovableHandle