torch.nn.modules.module.register_module_full_backward_pre_hook¶
- torch.nn.modules.module.register_module_full_backward_pre_hook(hook)[source][source]¶
注册适用于所有模块的向后预钩子。
警告
这将为 nn.module 模块添加全局状态，仅用于调试/性能分析。
使用此函数注册的钩子与使用
torch.nn.Module.register_full_backward_pre_hook()注册的钩子行为相同。请参阅其文档以获取更多详细信息。使用此函数注册的钩子将在使用
torch.nn.Module.register_full_backward_pre_hook()注册的钩子之前被调用。- 返回值:
一个可以用来通过调用
handle.remove()移除已添加钩子的句柄。- 返回类型:
torch.utils.hooks.RemovableHandle
