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

返回

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

返回类型

torch.utils.hooks.RemovableHandle