torch.nn.modules.module.register_module_backward_hook

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

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

此函数已被torch.nn.modules.module.register_module_full_backward_hook()替代,并且未来版本中该函数的行为将发生变化。

返回值

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

返回类型

torch.utils.hooks.RemovableHandle

本页目录