cr.nimble.dsp.nonzero_indices¶
- cr.nimble.dsp.nonzero_indices(x)[source]¶
Returns the indices of non-zero entries in x
- Parameters
x (jax.numpy.ndarray) – A sparse signal
- Returns
The indices of nonzero entries in x
- Return type
(jax.numpy.ndarray)
Note
This function cannot be JIT compiled as the size of output is data dependent.
See also