jax.Array.nonzero# abstract Array.nonzero(*, fill_value=None, size=None)[source]# Return indices of nonzero elements of an array. Refer to jax.numpy.nonzero() for the full documentation. Parameters: self (Array) fill_value (None | ArrayLike | tuple[ArrayLike, ...]) size (int | None) Return type: tuple[Array, …]
