torch.nn.modules.module.register_module_full_backward_pre_hook#
- torch.nn.modules.module.register_module_full_backward_pre_hook(hook)[源代码]#
为所有模块注册一个通用的 backward pre-hook。
警告
此函数会向 nn.module 模块添加全局状态，仅用于调试/剖析目的。
使用此函数注册的 hook 的行为与使用
torch.nn.Module.register_full_backward_pre_hook()注册的 hook 的行为相同。有关更多详细信息，请参阅其文档。使用此函数注册的 hook 将在调用使用
torch.nn.Module.register_full_backward_pre_hook()注册的 hook 之前被调用。- 返回
一个句柄，可用于通过调用
handle.remove()来移除添加的钩子- 返回类型
torch.utils.hooks.RemovableHandle
