def if_one(a: Tensor,
    b: Tensor) -> Tensor:
  if bool(torch.lt(a, b)):
    c = a
  else:
    c = b
  return c