torch.nn.modules.module.register_module_full_backward_hook

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

为所有模块注册一个共用的反向钩子。

警告

这在nn.module模块中添加了全局状态，仅用于调试和性能分析。

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

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