如何为 boost multi_array 动态设置维度

Question

我有一个函数需要创建一个 boost multi_array，其维度是先验未知的。但是，对于任何给定的维度，范围都是已知的。如何逐步为我的数组构建一个 extents 对象？

类似：

array_type::extent_gen<array_type::dimensionality> my_extents;
for (size_type d = 0; d < array_type::dimensionality; d++) {
    extents = extents[5];
}
array_type my_array(extents);

将一个尚未确定的数组设置为每个维度的范围为 5....

Answer 1

我不认为你可以，因为 multi_array 的维度是在类型的模板参数中编码的：

typedef boost::multi_array<double, 3> array_type;

从根本上不同于

typedef boost::multi_array<double, 2> array_type;

你可能做的“最好”就是拥有

static const int MAX_DIMENSIONS = 10;
typedef boost::multi_array<double, MAX_DIMENSIONS> array_type;

然后构建范围以将“未使用”维度的范围留在1：

Live On Coliru

#include <boost/multi_array.hpp>
#include <iostream>

static const int MAX_DIMENSIONS = 5;
typedef boost::multi_array<double, MAX_DIMENSIONS> array_type;

array_type make_array(int dims, int default_extent = 5) {
    boost::array<int, MAX_DIMENSIONS> exts = {};
    std::fill_n(exts.begin(), exts.size(), 1);
    std::fill_n(exts.begin(), dims, default_extent);

    return array_type(exts);
}

int main() {
    auto a = make_array(3);
    auto b = make_array(2);

    std::copy(a.shape(), a.shape() + a.dimensionality, std::ostream_iterator<size_t>(std::cout << "\na: ", " "));
    std::copy(b.shape(), b.shape() + b.dimensionality, std::ostream_iterator<size_t>(std::cout << "\nb: ", " "));
}

打印

a: 5 5 5 1 1 
b: 5 5 1 1 1