Function topk_axis

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

Returns the k largest elements from the input along a given axis.

The elements will not necessarily be in sorted order.

See [topk_all] for the flattened array.

Params

• a: The input array.
• k: The number of elements to return.
• axis: Axis to sort over. Default to -1 if not specified.