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