zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-21 10:18:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Minor updates to address a few issues (#537)

Browse Source 
* docs on arg indices return type

* arange with nan

* undo isort
This commit is contained in:
Awni Hannun
2024-01-23 22:24:41 -08:00
committed by GitHub
parent 4fe2fa2a64
commit f30e63353a
3 changed files with 23 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/src/ops.cpp
View File

@@ -2254,7 +2254,7 @@ void init_ops(py::module_& m) {
singleton dimensions, defaults to `False`.
Returns:
array: The output array with the indices of the minimum values.
array: The ``uint32`` array with the indices of the minimum values.
)pbdoc");
m.def(
"argmax",
@@ -2287,7 +2287,7 @@ void init_ops(py::module_& m) {
singleton dimensions, defaults to `False`.
Returns:
array: The output array with the indices of the maximum values.
array: The ``uint32`` array with the indices of the maximum values.
)pbdoc");
m.def(
"sort",
@@ -2343,7 +2343,7 @@ void init_ops(py::module_& m) {
If unspecified, it defaults to -1 (sorting over the last axis).
Returns:
array: The indices that sort the input array.
array: The ``uint32`` array containing indices that sort the input.
)pbdoc");
m.def(
"partition",
@@ -2416,7 +2416,7 @@ void init_ops(py::module_& m) {
If unspecified, it defaults to ``-1``.
Returns:
array: The indices that partition the input array.
array: The `uint32`` array containing indices that partition the input.
)pbdoc");
m.def(
"topk",