torch.nn.modules.module.register_module_backward_hook#
- torch.nn.modules.module.register_module_backward_hook(hook)[source]#
注册所有模块通用的后向钩子。
此函数已被弃用,建议使用
torch.nn.modules.module.register_module_full_backward_hook()
,并且此函数在未来版本中的行为可能会发生变化。- 返回
一个句柄,可用于通过调用
handle.remove()
来移除添加的钩子- 返回类型
torch.utils.hooks.RemovableHandle