Add CuPy signature for partition

kgryte · kgryte · commit fade09ebbc83 · 2023-12-14T02:06:42.000-08:00
diff --git a/signatures/searching/top_k.md b/signatures/searching/top_k.md
@@ -12,7 +12,7 @@ numpy.topk(a, k, axis=-1, largest=True, sorted=True) → [ndarray, ndarray]
 numpy.partition(a, kth, axis=-1, kind='introselect', order=None) → ndarray
 ```
 
-**Note**: returns an array of the same shape as `a` and requires sorting the return value to get the top `k` values in order.
+**Note**: returns an array of the same shape as `a` and requires sorting the return value to get the top `k` values in order. If `axis` is `None`, partitions a flattened array.
 
 ```
 numpy.argpartition(a, kth, axis=-1, kind='introselect', order=None) → ndarray
@@ -28,6 +28,10 @@ cupy.argpartition(a, kth, axis=-1) → ndarray
 
 **Note**: performs a full sort.
 
+```
+cupy.partition(a, kth, axis=-1) → ndarray
+```
+
 ## dask.array
 
 ```
