CountNonzero ============ Return the number of nonzero elements along a given axis. **Abstract Signature:** ``CountNonzero(a: Tensor, axis: int | tuple | None, keepdims: bool = False)`` .. raw:: html

PyTorch

API: torch.count_nonzero
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.count_nonzero
Strategy: Direct Mapping

NumPy

API: numpy.count_nonzero
Strategy: Direct Mapping

Keras

API: keras.ops.count_nonzero
Strategy: Direct Mapping

TensorFlow

API: tf.math.count_nonzero
Strategy: Direct Mapping