pytorch / 2 / generated / torch.nn.modules.module.register_module_backward_hook.html

torch.nn.modules.module.register_module_backward_hook

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

Registers a backward hook common to all the modules.

This function is deprecated in favor of torch.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 calling handle.remove()

Return type

torch.utils.hooks.RemovableHandle

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.modules.module.register_module_backward_hook.html