Function float_sort_with_indices 

pub fn float_sort_with_indices(
    tensor: HostTensor,
    dim: usize,
    descending: bool,
    indices_dtype: IntDType,
) -> (HostTensor, HostTensor)