jax.numpy.flatnonzero ===================== .. currentmodule:: jax.numpy .. autofunction:: flatnonzero