cr.nimble.dsp.largest_indices¶
- cr.nimble.dsp.largest_indices(x, K)[source]¶
Returns the indices of K largest entries in x by magnitude
- Parameters
x (jax.numpy.ndarray) – An data vector/point
K (int) – The number of largest entries to be identified in x
- Returns
An index vector of size K identifying the K largest entries in x in descending order
- Return type
(jax.numpy.ndarray)