Function sort_all_device

pub fn sort_all_device(
    a: impl AsRef<Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>

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

Params

• array: input array

Example

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

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