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