real and imag properties (#2189)

2025-10-19 08:38:09 +08:00 · 2025-05-15 18:17:50 -07:00
parent c1eb9d05d9
commit a2cadb8218
3 changed files with 23 additions and 0 deletions
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -319,6 +319,18 @@ void init_array(nb::module_& m) {
          R"pbdoc(
            The array's :class:`Dtype`.
          )pbdoc")
+      .def_prop_ro(
+          "real",
+          [](const mx::array& a) { return mx::real(a); },
+          R"pbdoc(
+            The real part of a complex array.
+          )pbdoc")
+      .def_prop_ro(
+          "imag",
+          [](const mx::array& a) { return mx::imag(a); },
+          R"pbdoc(
+            The imaginary part of a complex array.
+          )pbdoc")
      .def(
          "item",
          &to_scalar,