2

我需要将一个 multi_array 的大小调整为另一个的大小。

在 Blitz++ 中，我可以做到

arr1.resize(arr2.shape());

是否有类似长度的 multi_array 解决方案？因为

arr1.resize(boost::extents[arr2.shape()[0]][arr2.shape()[1]]);

似乎有点漫长和艰辛。

4

2 回答 2

1

您可以使用shape()会员。可悲的是，它不能直接作为一个ExtentList（它没有模拟这个Collection概念），但很容易把它变成一个：

using MA = multi_array<double, 2>;
MA ma(extents[12][34]);
auto& ma_shape = reinterpret_cast<boost::array<size_t, MA::dimensionality> const&>(*ma.shape());

以便

// demo
std::cout << "[" << ma_shape[0] << "][" << ma_shape[1] << "]\n";

打印[12][34]。

现在，ma_shape可以直接用于重塑/调整另一个数组的大小：

Live On Coliru

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

int main() {
    using namespace boost;
    using MA = multi_array<double, 2>;

    MA ma(extents[12][34]);
    auto& ma_shape = reinterpret_cast<boost::array<size_t, MA::dimensionality> const&>(*ma.shape());

    // demo
    std::cout << "[" << ma_shape[0] << "][" << ma_shape[1] << "]\n";

    // resize
    MA other;
    assert(!std::equal(ma_shape.begin(), ma_shape.end(), other.shape()));

    other.resize(ma_shape);
    assert(std::equal(ma_shape.begin(), ma_shape.end(), other.shape()));

    // reshape
    other.resize(extents[1][12*34]);
    assert(!std::equal(ma_shape.begin(), ma_shape.end(), other.shape()));

    other.reshape(ma_shape);
    assert(std::equal(ma_shape.begin(), ma_shape.end(), other.shape()));
}
于 2015-05-06T22:02:10.760 回答
0

我认为这是对 Boost.MultiArray 的另一个疏忽。我已经编写了一堆“实用”函数，它们允许采用形状（维度大小）、基本索引（例如，每个维度为 0 或 1）和扩展（每个维度中的基本索引和大小）。

namespace boost{
template<class MultiArray>
detail::multi_array::extent_gen<MultiArray::dimensionality>
extension(MultiArray const& ma){ //this function is adapted from 
    typedef detail::multi_array::extent_gen<MultiArray::dimensionality> gen_type;
    gen_type ret;
    typedef typename gen_type::range range_type;
    for(int i=0; i != MultiArray::dimensionality; ++i)
        ret.ranges_[i] = range_type(ma.index_bases()[i], ma.index_bases()[i]+ma.shape()[i]);
    return ret;
}
}

后来用作：

boost::multi::array<double, 3> m(boost::multi::extents[3][4][5]);
boost::multi::array<double, 3> n(extension(m)); // n takes the extension (and shape) of m

（如果基本索引不为零，它也可以工作。它也适用于视图和其他 multi_array 类型的类。）

于 2015-07-02T18:21:51.857 回答