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