Function adjust_strided_shape

src/shape_transform_descriptor.cpp:2067–2091

}

static std::size_t adjust_strided_shape(shape& s, std::size_t n)
{
    auto lens    = s.lens();
    auto strides = s.strides();

    // Insert a dim of 1 so it can be used to handle steps
    if(std::none_of(strides.begin(), strides.end(), [](auto stride) { return stride == 1; }) and
       std::any_of(strides.begin(), strides.end(), [](auto stride) { return stride != 0; }))
    {
        lens.push_back(1);
        strides.push_back(1);
    }

    auto last_axis      = std::max_element(strides.begin(), strides.end()) - strides.begin();
    auto total_elements = std::max<std::size_t>(1, strides[last_axis] * lens[last_axis]);
    // Add a dim of 1 to the front so it can handle extra elements
    auto extra = n / total_elements;
    if(extra > 1)
    {
        strides.insert(strides.begin(), total_elements);
        lens.insert(lens.begin(), 1);
    }
    s = shape(s.type(), lens, strides);
    return std::max<std::size_t>(1, extra);
}

template <class Range>
static std::vector<std::size_t> select_mask(const std::vector<std::size_t>& slice_mask,