c++ расплющить вложенный цикл?

Как-то так: https://gcc.godbolt.org/z/5GsYbjK3j

Как я понимаю, N - может меняться. У меня вышло что-то такое (без cppcoro): https://godbolt.org/z/bMEEMW7bM

Пока без compile-time рекурсии не смог обойтись.

Поправил пост. Без рантайм-реакурсии, конечно, как оно при компиляции будет идти вглубь — не важно.

Ну вот вообще без рекурсии:

namespace flatten1 {
template<size_t N>
generator<int> flatten(int *shape)
{
    std::array<int, N> i{0};
    int n = 0;
    bool process = true;

    while (process) {

        for (i.back() = 0; i.back() < shape[N-1]; ++i.back()) {
            n++;
            co_yield n;
        }

        if (N > 1) {
            for (auto j = N-1; j >= 1; --j) {
                auto const index = j - 1;
                if (++i[index] == shape[index]) {
                    if (index == 0) {
                        process = false;
                        break;
                    }
                    i[index] = 0;
                    continue;
                }
                break;
            }
        }
    }
}
}

Я не помню, но std::array<int, N> нормально создатся на стеке.

Или вот вариант с «генерацией» шейпа, по типу как у rupert:

namespace flatten2 {
template<size_t N>
generator<std::array<int, N>> flatten(int *shape)
{
    std::array<int, N> i{0};

    bool process = true;
    int start = 1; // hack, to avoid modification generator<>::promise_type::initial_suspend() to std::suspend_never()

    while (process) {

        for (i.back() = start; i.back() < shape[N-1]; ++i.back()) {
            co_yield i;
        }

        if (N > 1) {
            for (auto j = N-1; j >= 1; --j) {
                auto const index = j - 1;
                if (++i[index] == shape[index]) {
                    if (index == 0) {
                        process = false;
                        break;
                    }
                    i[index] = 0;
                    continue;
                }
                break;
            }
        }

        start = 0;
    }
}
}

Всё вместе и подправленное: https://godbolt.org/z/bMEEMW7bM

Shape вынеси в класс, добавить итератор который будет перескакивать границы.

потом так

for(auto coord = shape.coordBegin(); coord.isValid(); ++coord)

Если кусок памяти монолитный (т.е. между размерностями нет страйдов и прочих выравниваний), то

for (int idx = 0; idx < dims[0]*dims[1]*dims[2]*dims[3]; idx++) {
    ...
}

Как на шаблонах перемножить все элементы массива shape известной размерности, думаю сами справитесь))

Все зависит от:

1. того что хочется в этом цикле делать.

2. хочется ли его делать параллельным.

3. насколько тело цикла толстое (насколько критичны накладные расходы).

Варианты обственно такие:

1. делаем один целочисленный счетчик по всей области и на каждой итерации переводим его в j0…3. Плюсы - это просто и хорошо параллелится. Минусы - это дорого (лишние телодвижения), но при условии что нужны j0…3. Если не нужны то это самый быстрый варинат.

2. Делаем счетчик в виде n целых чисел, инкрементируем первое а дальше проверяем границы. Плюсы - это быстро (нет лишних телодвижений), минусы - это не параллелится и надо писать всякие обертки что бы это нормально смотрелось.

3. Если нужна параллельность - берем в.2, но не инкрементируем первую компоненту а добавляем число тредов, дальше примерно то же самое. В использовании это выглядит странновато, хотя привыкнуть можно.

Если нужны готовые решения их есть у меня (собственно мы этими многомерными циклами дцать лет пользуемся), могу кинуть в почту ссылку на гитхаб.

Всё разобрался. Я в постановке перемудрил — оказалось, что итератор не нужен, достаточно рекурсивного шаблона с коллбэк-лямбдой, а в неё уже можно напихать всего при необходимости.

Жуткая жуть….

достаточно рекурсивного шаблона с коллбэк-лямбдой

Какой ты всё-таки скучный). Тут тебе таких решений накидали, а ты!

Кстати, по поводу коллбэк-лямбды: https://github.com/arximboldi/zug, https://sinusoid.es/zug/transducer.html#zip
Я думаю, что можно zip-нуть сразу несколько shape диапазонов, и потом по этому зипу пройтись, как по простому итератору туплов.

Какой ты всё-таки скучный). Тут тебе таких решений накидали, а ты!

Да ваще :)

Но на самом деле, я тут разбирался с корутинами, вот и :)

Тут не хватает выворачиваний циклов на изнанку, Дафф-девайсов и «анроллов» через TBB и OMP и прочие параллел-форы :)

а самый рофл что для поэлементного прохода ничего из этого не нужно

мсьи должны все это применить, особенно там где не нужно :)

За метапрограммирование на шаблонах на работе разве руки не отрезают?

тихо, не спугни