blob: 1dff15859b137781b5b5991627ae02664e7a7802 [file] [log] [blame]
def if_test(a: Tensor,
b: Tensor) -> Tensor:
if bool(torch.lt(a, b)):
c = b
else:
c = a
return c