Flatnonsense

Return indices of nonzero elements in a flattened array

Abstract Signature:

Flatnonsense(a: Tensor, size: int | None, fill_value = 0)

JAX (Core)

API: jax.numpy.flatnonzero
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.flatnonzero
Strategy: Direct Mapping
Official Docs ↗