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

torch.nn.modules.module.register_module_full_backward_pre_hook

torch.nn.modules.module.register_module_full_backward_pre_hook(hook)[源代码]

注册一个通用于所有模块的反向预钩子。

警告

这为nn.module模块添加了全局状态,仅用于调试/性能分析目的。

使用此函数注册的钩子与通过 torch.nn.Module.register_full_backward_pre_hook() 注册的钩子行为相同。有关更多详细信息,请参阅其文档。

使用此函数注册的钩子将在使用torch.nn.Module.register_full_backward_pre_hook()注册的钩子之前被调用。

Returns

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

Return type

torch.utils.hooks.RemovableHandle

优云智算