Table of Contents
Shortcuts

# torch.nn.modules.module.register_module_full_backward_hook

torch.nn.modules.module.register_module_full_backward_hook(hook)

  Registers a backward hook common to all the modules.

  Warning

  This adds global state to the nn.module module and it is only intended for debugging/profiling purposes.

  The hook will be called every time the gradients with respect to a module are computed, i.e. the hook will execute if and only if the gradients with respect to module outputs are computed. The hook should have the following signature:

      hook(module, grad_input, grad_output) -> Tensor or None

  The `grad_input` and `grad_output` are tuples. The hook should not modify its arguments, but it can optionally return a new gradient with respect to the input that will be used in place of `grad_input` in subsequent computations. `grad_input` will only correspond to the inputs given as positional arguments and all kwarg arguments will not appear in the hook. Entries in `grad_input` and `grad_output` will be `None` for all non-Tensor arguments.

  For technical reasons, when this hook is applied to a Module, its forward function will receive a view of each Tensor passed to the Module. Similarly the caller will receive a view of each Tensor returned by the Module’s forward function.

  Global hooks are called before hooks registered with register_backward_hook

  Returns:

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

  Return type:

    `torch.utils.hooks.RemovableHandle`

## Docs

Access comprehensive developer documentation for PyTorch

View Docs

## Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

## Resources

Find development resources and get your questions answered

View Resources

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.