我有一个n维Boost.MultiArray,我初始化如下:
const int n=3, size=4; //# of dimensions and size of one dimension
boost::multi_array<char,n> arr;
boost::array<size_t,n> extents; //size of each dimension
extents.assign(size); //assign size to each dimension -> {{4, 4, 4}}
arr.resize(extents);
所以我有 4 行代码来获取 MultiArray,但我想用一行代码来完成。 有没有简单的方法来生成一个具有 n 维的 MultiArray,每个维都有
size
长度(所以我可以写 arr(samevaluearray(n,size))
)或者我是否错过了一个方便的 MultiArray 构造函数?
编辑: 它应该在不依赖于 n 的特定值的情况下工作,即
arr({{size,size}}
只适用于 n=2
。
因为可能不清楚:
boost::multi_array<char,n>(boost::extents[4][4][4])
正确地初始化了4x4x4数组,但是每次在源代码中更改n
时,每个初始化都必须手动更新,所以这不是一个选项。
您可以将数组的创建封装到辅助函数中:
template <typename T, size_t N>
boost::multi_array<T, N> make_regular_matrix(const size_t m)
{
boost::multi_array<T, N> arr;
boost::array<size_t, N> extents;
extents.assign(m);
arr.resize(extents);
return arr;
}
const int n = 3;
int size = 4; // Can be const as well, but this is not mandatory
auto arr = make_regular_matrix<char, n>(size);
如果无法使用
auto
,则必须复制模板参数:
boost::multi_array<char, n> arr = make_regular_matrix<char, n>(size);
make_regular_matrix
函数可以缩短为使用std::vector
,就像您在答案中所做的那样;我不知道这样的实现是否会更好。辅助函数的目的是隐藏数组的创建,但可以编写其他版本,例如用给定值初始化数组元素:
template <size_t N, typename T> //switched order for deduction
boost::multi_array<T, N> make_regular_matrix(const size_t m, const T & value)
{
boost::multi_array<T, N> arr(std::vector<size_t>(n, m));
std::fill(arr.data(), arr.data() + arr.num_elements(), value);
return arr;
}
auto arr = make_regular_matrix<4>(3, 'z'); //creates a 3x3x3x3 matrix
//filled with 'z's
事实证明,
std::vector
有一个构造函数,它构造一个具有重复n次的常量值的向量,因此可能的解决方案如下所示:
const int n=2, size=4; //# of dimensions and size of one dimension
boost::multi_array<char,n> arr(std::vector<size_t>(n,size));
这会初始化一个 n 维 multi_array,每个维度的大小设置为 size。
从 Boost Multi-Array 文档,是的,你可以用一行初始化它:
typedef boost::multi_array<double, 3> array_type;
typedef array_type::index index;
array_type A(boost::extents[3][4][2]);
typedef 是为了提高可读性,您可以轻松地为您的示例做到这一点:
boost::multi_array<int, 2> arr(boost::extents[2][4]);
我想说这里的问题实际上是如何简单地生成具有单个值的常量静态数组。
template<std::size_t N>
std::array<size_t, N> make_regular_extents(std::size_t m) {
std::array<size_t, N> extents; // can be boost::array
extents.fill(m); // can be .assign for boost::array
return extents;
}
int main() {
const int n = 3;
int size = 4; // Can be const as well, but this is not mandatory
boost::multi_array<char, n> arr(make_regular_extents<n>(size));
}
或一行(根据要求):
boost::multi_array<char, n> arr2(
std::apply([size](auto... e) {return std::array{(e, size)...};}, std::array<int, n>{})
);