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() 注册的钩子之前调用。

返回

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

返回类型

torch.utils.hooks.RemovableHandle