Function af_imag

src/api/c/complex.cpp:124–150

}

af_err af_imag(af_array *out, const af_array in) {
    try {
        const ArrayInfo &info = getInfo(in);
        af_dtype type         = info.getType();

        if (type != c32 && type != c64) {
            return af_constant(out, 0, info.ndims(), info.dims().get(), type);
        }
        if (info.ndims() == 0) { return af_retain_array(out, in); }

        af_array res;
        switch (type) {
            case c32:
                res = getHandle(imag<float, cfloat>(getArray<cfloat>(in)));
                break;
            case c64:
                res = getHandle(imag<double, cdouble>(getArray<cdouble>(in)));
                break;

            default: TYPE_ERROR(0, type);
        }

        std::swap(*out, res);
    }
    CATCHALL;
    return AF_SUCCESS;
}

af_err af_conjg(af_array *out, const af_array in) {
    try {