torch.nn.modules.module.register_module_backward_hook — PyTorch 2.7 documentation (original) (raw)

torch.nn.modules.module.register_module_backward_hook(hook)[source][source]¶

Register a backward hook common to all the modules.

This function is deprecated in favor oftorch.nn.modules.module.register_module_full_backward_hook()and the behavior of this function will change in future versions.

Returns

a handle that can be used to remove the added hook by callinghandle.remove()

Return type

torch.utils.hooks.RemovableHandle