NonZero ======= Returns the indices of the elements that are non-zero (in row-major order - by dimension). NonZero behaves similar to numpy.nonzero: https://docs.scipy.org/doc/numpy/reference/generated/numpy.nonzero.html, but for scalar input, NonZero produces output shape (0, N) instead of (1, N), which is differe... **Abstract Signature:** ``NonZero(X: Tensor)`` .. raw:: html

JAX (Core)

API: jax.numpy.nonzero
Strategy: Direct Mapping

NumPy

API: numpy.ma.core.nonzero
Strategy: Direct Mapping

Keras

API: keras.ops.numpy.nonzero
Strategy: Direct Mapping

TensorFlow

API: keras.ops.nonzero
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.nonzero
Strategy: Direct Mapping