• Docs >
  • torch.nn >
  • torch.nn.modules.module.register_module_backward_hook
Shortcuts

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() ,此函数的行为将在未来版本中发生变化。

Returns

一个可以用于通过调用handle.remove()来移除添加的钩子的句柄

Return type

torch.utils.hooks.RemovableHandle