Function sort 

pub fn sort<T, D>(
    a: &Array<T, D>,
    axis: Option<usize>,
    kind: SortKind,
) -> FerrayResult<Array<T, D>>
where
    T: Element + PartialOrd + Copy + Send + Sync,
    D: Dimension,

Sort an array along the given axis (or flattened if axis is None).

Returns a new sorted array with the same shape.

Equivalent to numpy.sort.