Real and Imag (#1490)

* real and imag * fix * fix
2025-12-16 01:49:05 +08:00 · 2024-10-15 16:23:15 -07:00
parent 2b8ace6a03
commit 3f86399922
21 changed files with 275 additions and 46 deletions
--- a/mlx/ops.h
+++ b/mlx/ops.h
@@ -1480,6 +1480,12 @@ array roll(
    const std::vector<int>& axes,
    StreamOrDevice s = {});

+/* The real part of a complex array. */
+array real(const array& a, StreamOrDevice s = {});
+
+/* The imaginary part of a complex array. */
+array imag(const array& a, StreamOrDevice s = {});
+
 /** @} */

 } // namespace mlx::core