cr.nimble.dsp.support

cr.nimble.dsp.support(x)[source]

Returns the indices of non-zero entries in x

Parameters

x (jax.numpy.ndarray) – A sparse signal

Returns

The support of x a.k.a. 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.