Function f

intermediate_source/autograd_saved_tensors_hooks_tutorial.py:95–96

(x)

#

def f(x):
    return x * x

x = torch.randn(5, requires_grad=True)
y = f(f(f(x)))