cr.nimble.dsp.sparse_approximation_rw

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

Keeps only largest K non-zero entries by magnitude in each row of X

Parameters

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

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

Returns

X modified so that all entries except the K largest entries are set to 0 in each row

Return type

(jax.numpy.ndarray)