mlx_rs::ops::indexing

Function topk_all

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

Returns the k largest elements from the flattened input array.