Function foo

intermediate_source/torch_compile_tutorial.py:73–76

(x, y)

def foo(x, y):
    a = torch.sin(x)
    b = torch.cos(y)
    return a + b


opt_foo1 = torch.compile(foo)