Function argsort_device

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

Returns the indices that sort the array.

See argsort_all for the flattened array.

Params

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