Add conjugate operator (#1100)

* cpu and gpu impl

* add mx.conj and array.conj()

---------

Co-authored-by: Alex Barron <abarron22@apple.com>
This commit is contained in:
Alex Barron
2024-05-10 07:22:20 -07:00
committed by GitHub
parent 8bd6bfa4b5
commit 2e158cf6d0
17 changed files with 143 additions and 11 deletions

View File

@@ -1584,5 +1584,13 @@ void init_array(nb::module_& m) {
"stream"_a = nb::none(),
R"pbdoc(
Extract a diagonal or construct a diagonal matrix.
)pbdoc");
)pbdoc")
.def(
"conj",
[](const array& a, StreamOrDevice s) {
return mlx::core::conjugate(to_array(a), s);
},
nb::kw_only(),
"stream"_a = nb::none(),
"See :func:`conj`.");
}