Function sort 

pub fn sort(tensor: HostTensor, dim: usize, descending: bool) -> HostTensor

Sort elements along a dimension, returning the sorted tensor.