Function true_branch

intermediate_source/torch_compile_tutorial.py:375–376

(y)

    x = a / (torch.abs(a) + 1)

    def true_branch(y):
        return y * -1

    def false_branch(y):
        # NOTE: torch.cond doesn't allow aliased outputs