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

add std as method (#1387)

Browse Source 
* add std as method

* add std as method
This commit is contained in:
Awni Hannun 2024-09-01 19:49:16 -07:00 committed by GitHub
parent 58dca7d846
commit 9592766939
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 19 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/python/array.rst
View File

@ -53,8 +53,9 @@ Array
array.sqrt array.sqrt
array.square array.square
array.squeeze array.squeeze
array.swapaxes array.std
array.sum array.sum
array.swapaxes
array.transpose array.transpose
array.T array.T
array.var array.var

1
docs/src/python/ops.rst
View File

@ -78,6 +78,7 @@ Operations
hadamard_transform hadamard_transform
identity identity
inner inner
isfinite
isclose isclose
isinf isinf
isnan isnan

16
python/src/array.cpp
View File

@ -1115,6 +1115,22 @@ void init_array(nb::module_& m) {
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
"See :func:`mean`.") "See :func:`mean`.")
.def(
"std",
[](const array& a,
const IntOrVec& axis,
bool keepdims,
int ddof,
StreamOrDevice s) {
return mlx::core::std(
a, get_reduce_axes(axis, a.ndim()), keepdims, ddof, s);
},
"axis"_a = nb::none(),
"keepdims"_a = false,
"ddof"_a = 0,
nb::kw_only(),
"stream"_a = nb::none(),
"See :func:`std`.")
.def( .def(
"var", "var",
[](const array& a, [](const array& a,