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