如何将元素插入到具有动态大小的ublas矩阵中

时间:2014-12-31 09:41:14

标签: c++ boost-ublas

尝试在最初未指定大小的矩阵中插入元素时出现运行时错误。

以下代码为m1运行finr,但抛出 m2 的错误。

#include <iostream>
#include <boost/numeric/ublas/matrix.hpp>
#include <boost/numeric/ublas/io.hpp>    

int main () {
    boost::numeric::ublas::matrix<double> m1 (1,1);
    boost::numeric::ublas::matrix<double> m2;
    unsigned int i = 0;
    m1(i,i)=9;  // This works completely
    m2(i,i)=9;  // This one throws error
    return 0;
}

如果有人可以提供替代方案来实现这一点,我将非常感激。

1 个答案:

答案 0 :(得分:2)

正如您已经注意到的那样,boost::numeric::ublas::matrix并未自动调整自身大小,例如std::vector。您必须在调用operator()之前手动调整大小,或者编写一个为您调整大小的函数模板,如下所示:

namespace ublas = boost::numeric::ublas; //just a friendly alias!

template<typename T, typename U>
void assign(ublas::matrix<T>& m,std::size_t r,std::size_t c,U const& data)
{
    m.resize(std::max(m.size1(), r+1), std::max(m.size2(), c+1));
    m(r, c) = data;
}

现在你可以这样做:

int main () 
{
    ublas::matrix<double> m;

    assign(m, 0, 0, 9);  //m(0,0) = 9; 
    assign(m, 3, 2, 20); //m(3,2) = 20

    std::cout << m << std::endl; //prints: [4,3]((9,0,0),(0,0,0),(0,0,0),(0,0,20))
    return 0;
}

Online demo

希望有所帮助。