Function seqToDims

src/api/cpp/array.cpp:71–98

}

af::dim4 seqToDims(af_index_t *indices, af::dim4 parentDims,
                   bool reorder = true) {
    try {
        af::dim4 odims(1);
        for (int i = 0; i < AF_MAX_DIMS; i++) {
            if (indices[i].isSeq) {
                odims[i] = calcDim(indices[i].idx.seq, parentDims[i]);
            } else {
                dim_t elems = 0;
                AF_THROW(af_get_elements(&elems, indices[i].idx.arr));
                odims[i] = elems;
            }
        }

        // Change the dimensions if inside GFOR
        if (reorder) {
            for (int i = 0; i < AF_MAX_DIMS; i++) {
                if (indices[i].isBatch) {
                    int tmp  = odims[i];
                    odims[i] = odims[3];
                    odims[3] = tmp;
                    break;
                }
            }
        }
        return odims;
    } catch (const logic_error &err) { AF_THROW_ERR(err.what(), AF_ERR_SIZE); }
}

unsigned numDims(const af_array arr) {
    unsigned nd;