Function af_arg

src/api/c/unary.cpp:618–642

}

af_err af_arg(af_array *out, const af_array in) {
    try {
        const ArrayInfo &in_info = getInfo(in);
        if (in_info.ndims() == 0) { return af_retain_array(out, in); }

        if (!in_info.isComplex()) {
            return af_constant(out, 0, in_info.ndims(), in_info.dims().get(),
                               in_info.getType());
        }

        af_array real;
        af_array imag;

        AF_CHECK(af_real(&real, in));
        AF_CHECK(af_imag(&imag, in));

        AF_CHECK(af_atan2(out, imag, real, false));

        AF_CHECK(af_release_array(real));
        AF_CHECK(af_release_array(imag));
    }
    CATCHALL;

    return AF_SUCCESS;
}

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