pytorch

1
def 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