1def if_test(a: Tensor, 2 b: Tensor) -> Tensor: 3 if bool(torch.lt(a, b)): 4 c = b 5 else: 6 c = a 7 return c 8