jax.Array.nonzero#

abstractArray.nonzero(*,fill_value=None,size=None)[source]#

Return indices of nonzero elements of an array.

Refer tojax.numpy.nonzero() for the full documentation.

Parameters:

• self (Array)

• fill_value (None |ArrayLike |tuple[ArrayLike,...])

• size (int |None)

Return type:

tuple[Array, …]