• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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