caldera.utils.stable_arg_sort_long

caldera.utils.stable_arg_sort_long(arr)

Stable sort of long tensors.

Note that Pytorch 1.5.0 does not have a stable sort implementation. Here we simply add a delta value between 0 and 1 (exclusive) and assuming we are using integers, call torch.argsort to get a stable sort.