Function unbroadcast

src/shape_transform_descriptor.cpp:2045–2065

Replace broadcasted dimensions with size 1, and set the stride to the previous stride

// Replace broadcasted dimensions with size 1, and set the stride to the previous stride
static shape unbroadcast(const shape& s)
{
    std::vector<std::size_t> lens    = s.lens();
    std::vector<std::size_t> strides = s.strides();
    auto stride_it                   = std::find_if(
        s.strides().begin(), s.strides().end(), [](auto stride) { return stride != 0; });
    std::size_t prev_stride = stride_it == s.strides().end() ? 1 : *stride_it;
    for(std::size_t i = 0; i < lens.size(); ++i)
    {
        if(strides[i] == 0)
        {
            lens[i]    = 1;
            strides[i] = prev_stride;
        }
        else
        {
            prev_stride = strides[i];
        }
    }
    return {s.type(), lens, strides};
}

static std::size_t adjust_strided_shape(shape& s, std::size_t n)
{