Add signatures for count_nonzero

kgryte · kgryte · commit 01d47bde30f1 · 2024-04-18T03:31:19.000-07:00
diff --git a/signatures/statistics/count_nonzero.md b/signatures/statistics/count_nonzero.md
@@ -0,0 +1,49 @@
+# count_nonzero
+
+## NumPy
+
+```
+numpy.count_nonzero(a, axis=None, *, keepdims=False) → ndarray
+```
+
+## CuPy
+
+```
+cupy.count_nonzero(a, axis=None) → ndarray
+```
+
+No support for `keepdims`.
+
+## dask.array
+
+```
+dask.array.count_nonzero(a, axis=None) → ndarray
+```
+
+No support for `keepdims`.
+
+## JAX
+
+```
+jax.numpy.count_nonzero(a, axis=None, keepdims=False) → ndarray
+```
+
+## MXNet
+
+```
+np.count_nonzero(a, axis=None, *, keepdims=False) → ndarray
+```
+
+## PyTorch
+
+```
+torch.count_nonzero(input, dim=None) → Tensor
+```
+
+No support for `keepdims`.
+
+## TensorFlow
+
+```
+tf.math.count_nonzero(input, axis=None, keepdims=False, dtype=tf.dtypes.int64, name=None) → Tensor
+```
