mlx_rs::ops::indexing

Function argsort

Source
pub fn argsort(a: impl AsRef<Array>, axis: i32) -> Result<Array>
Expand description

Returns the indices that sort the array.

See argsort_all for the flattened array.

ยงParams

  • a: The input array.
  • axis: Axis to sort over.