Function af_mean_all_weighted

src/api/c/mean.cpp:187–229

}

af_err af_mean_all_weighted(double *realVal, double *imagVal, const af_array in,
                            const af_array weights) {
    try {
        const ArrayInfo &iInfo = getInfo(in);
        const ArrayInfo &wInfo = getInfo(weights);
        af_dtype iType         = iInfo.getType();
        af_dtype wType         = wInfo.getType();

        ARG_ASSERT(
            3,
            (wType == f32 ||
             wType ==
                 f64)); /* verify that weights are non-complex real numbers */

        switch (iType) {
            case f32:
            case s32:
            case u32:
            case s16:
            case u16:
            case s8:
            case u8:
            case b8:
            case f16: *realVal = mean<float>(in, weights); break;
            case f64:
            case s64:
            case u64: *realVal = mean<double>(in, weights); break;
            case c32: {
                cfloat tmp = mean<cfloat>(in, weights);
                *realVal   = real(tmp);
                *imagVal   = imag(tmp);
            } break;
            case c64: {
                cdouble tmp = mean<cdouble>(in, weights);
                *realVal    = real(tmp);
                *imagVal    = imag(tmp);
            } break;
            default: TYPE_ERROR(1, iType);
        }
    }
    CATCHALL;
    return AF_SUCCESS;
}