torch.nn.modules.module.register_module_full_backward_pre_hook

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

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

警告

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

使用此函数注册的 Hooks 的行为与通过 torch.nn.Module.register_full_backward_pre_hook() 注册的 Hooks 行为相同。更多详情请参阅相关文档。

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

返回值

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

返回类型

torch.utils.hooks.RemovableHandle

本页目录