Indexing.hpp

#pragma once
 
#include <cstddef>
 
#include <stratax/core/Shape.hpp>
 
namespace stratax::indexing {
 
template<typename IndexContainer>
std::size_t offset(
    const core::Shape& strides,
    const IndexContainer& indices)
{
    std::size_t result = 0;
 
    auto stride_it = strides.begin();
 
    for (const auto index : indices)
    {
        result += index * *stride_it;
        ++stride_it;
    }
 
    return result;
}
 
} // namespace stratax::indexing