add std as method (#1387)

* add std as method * add std as method
2025-10-31 07:58:14 +08:00 · 2024-09-01 19:49:16 -07:00
parent 58dca7d846
commit 9592766939
3 changed files with 19 additions and 1 deletions
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -1115,6 +1115,22 @@ void init_array(nb::module_& m) {
          nb::kw_only(),
          "stream"_a = nb::none(),
          "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(
          "var",
          [](const array& a,