Function af_moddims

src/api/c/moddims.cpp:42–84

}  // namespace

af_err af_moddims(af_array* out, const af_array in, const unsigned ndims,
                  const dim_t* const dims) {
    try {
        if (ndims == 0) {
            *out = retain(in);
            return AF_SUCCESS;
        }
        ARG_ASSERT(2, ndims >= 1);
        ARG_ASSERT(3, dims != NULL);

        af_array output = 0;
        dim4 newDims(ndims, dims);
        const ArrayInfo& info = getInfo(in);
        dim_t in_elements     = info.elements();
        dim_t new_elements    = newDims.elements();

        DIM_ASSERT(1, in_elements == new_elements);

        af_dtype type = info.getType();

        switch (type) {
            case f32: output = modDims<float>(in, newDims); break;
            case c32: output = modDims<cfloat>(in, newDims); break;
            case f64: output = modDims<double>(in, newDims); break;
            case c64: output = modDims<cdouble>(in, newDims); break;
            case b8: output = modDims<char>(in, newDims); break;
            case s32: output = modDims<int>(in, newDims); break;
            case u32: output = modDims<uint>(in, newDims); break;
            case s8: output = modDims<schar>(in, newDims); break;
            case u8: output = modDims<uchar>(in, newDims); break;
            case s64: output = modDims<intl>(in, newDims); break;
            case u64: output = modDims<uintl>(in, newDims); break;
            case s16: output = modDims<short>(in, newDims); break;
            case u16: output = modDims<ushort>(in, newDims); break;
            case f16: output = modDims<half>(in, newDims); break;
            default: TYPE_ERROR(1, type);
        }
        std::swap(*out, output);
    }
    CATCHALL

    return AF_SUCCESS;
}

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