Function foo3

intermediate_source/torch_compile_tutorial.py:146–150

(x)

def foo3(x):
    y = x + 1
    z = torch.nn.functional.relu(y)
    u = z * 2
    return u


opt_foo3 = torch.compile(foo3)