Function sort

pub fn sort(a: impl AsRef<Array>, axis: i32) -> Result<Array>

Returns a sorted copy of the array. Returns an error if the arguments are invalid.

Params

• array: input array
• axis: axis to sort over

Example

use mlx_rs::{Array, ops::*};

let a = Array::from_slice(&[3, 2, 1], &[3]);
let axis = 0;
let result = sort(&a, axis);