Function f1

intermediate_source/torch_compile_tutorial.py:238–241

(x, y)

def f1(x, y):
    if x.sum() < 0:
        return -y
    return y


# Test that `fn1` and `fn2` return the same result, given the same arguments `args`.