cr.nimble.dsp.largest_indices_cw

cr.nimble.dsp.largest_indices_cw(X, K)

Returns the indices of K largest entries by magnitude in each column of X

Parameters

• X (jax.numpy.ndarray) – An (N,S) data matrix with data points in columns

• K (int) – The number of largest entries to be identified in each column of X

Returns

An (K,S) index matrix indices of K largest elements in each column of X

Return type

(jax.numpy.ndarray)