Access real and imaginary parts of mx.complex64

[louisamand]

It seems it is not yet implemented (#361), but is there a workaround that doesn't force a conversion to NumPy ?

[angeloskath]

We should implement that for sure! In the meantime the following should work well

def real(x):
    return mx.view(x, mx.float32)[..., ::2]


def imag(x):
    return mx.view(x, mx.float32)[..., 1::2]

[louisamand]

Thanks! quite gracious :)