Function hook

intermediate_source/visualizing_gradients_tutorial.py:145–147

Forward pass hook which attaches backward pass hooks to intermediate tensors

(module, args, output)

def hook_forward(module_name, grads, hook_backward):
    def hook(module, args, output):
        """Forward pass hook which attaches backward pass hooks to intermediate tensors"""
        output.register_hook(hook_backward(module_name, grads))
    return hook

def hook_backward(module_name, grads):