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
torch.count_nonzero
jax.numpy.count_nonzero
numpy.count_nonzero
keras.ops.count_nonzero
tf.math.count_nonzero