C容器问题

2023年11月26日 286次阅读

我正在寻找一些合适的2D元素容器.我想要的是能够使用例如BOOST_FOREACH遍历容器的每个元素,我也希望能够构建我的容器的子视图(切片/子范围),并且可能也会迭代它们.

现在我正在使用boost :: numeric :: ublas :: matrix来实现这些目的,但是,它对我来说看起来不是一个好的解决方案,因为它是一个BLAS矩阵,尽管它表现得非常好一个简单的2d元素容器(自定义无界/有界存储也很甜).

另一个提升替代方法boost :: multi_array很糟糕,因为你不能使用一个BOOST_FOREACH语句迭代每个元素,因为构造视图的语法非常混乱.

任何替代品？

谢谢.

最佳答案 我执行以下操作(数组类型是容器/迭代器范围概念)：

ublas::matrix<douple> A;
foreach (double & element, A.data())
{
}

但是,这对切片不起作用：最好的解决方案是为它们编写迭代器.

以下是使用multi_array提供自定义类存储的示例.
也许你也可以这样做：

template<size_t N, typename T>
struct tensor_array : boost::multi_array_ref<T,N> {
    typedef boost::multi_array_ref<T,N> base_type;

    typedef T value_type;
    typedef T& reference;
    typedef const T& const_reference;

    tensor_array() : base_type(NULL, extents())
    {
        // std::cout << "create" << std::endl;
    }
    template<class A>
    tensor_array(const A &dims)
        : base_type(NULL, extents())
    {
        //std::cout << "create" << std::endl;
        resize(dims);
    }

    template<typename U>
    void resize(const U (&dims)[N]) {
        boost::array<U,N> dims_;
        std::copy(dims, dims + N, dims_.begin());
        resize(dims_);
    }

    template<typename U>
    void resize(const boost::array<U,N> &dims) {
        size_t size = 1;
        boost::array<size_t,N> shape;
        for (size_t i = 0; i < N; ++i)  {
            size *= dims[i];
            shape[N-(i+1)] = dims[i];
        }
        data_.clear();
        data_.resize(size, 0);
        // update base_type parent
        set_base_ptr(&data_[0]);
        this->num_elements_ = size;
        reshape(shape);
    }

    size_t size() const { return data_.size(); }
    size_t size(size_t i) const { return this->shape()[N-(i+1)]; }
    tensor_array& fill(const T &value) {
        std::fill(data_.begin(), data_.end(), value);
        return *this;
    }
private:
    typedef boost::detail::multi_array::extent_gen<N> extents;
    std::vector<T> data_;
};
点赞