Function opt_foo2

intermediate_source/torch_compile_tutorial.py:84–87

(x, y)

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


print(opt_foo2(torch.randn(3, 3), torch.randn(3, 3)))