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

@@ -4101,6 +4101,15 @@ array number_of_elements(
{a}));
}
array conjugate(const array& a, StreamOrDevice s /* = {} */) {
// Mirror NumPy's behaviour for real input
if (a.dtype() != complex64) {
return a;
}
return array(
a.shape(), a.dtype(), std::make_shared<Conjugate>(to_stream(s)), {a});
}
array bitwise_impl(
const array& a,
const array& b,