c++ - 关于 Armadillo 稀疏矩阵中的内存分配

Question

我想知道在创建稀疏矩阵后是否需要释放 locations 和 values 对象占用的内存。这是代码:

void load_data(umat &locations, vec& values){
  // fill up locations and values
}

int main(int argc, char**argv){
    umat loc;
    vec val;
    load_data(loc,val);
    sp_mat X(loc,val);
    return 0;
}

在上面的代码中，load_data() 填充了 locations 和 values 对象，然后在 main 中创建了稀疏矩阵()。我的问题:在构建X 后，是否需要释放位置和值使用的内存？原因是 X 可能很大，而我的内存不足。我知道当 main 返回时，操作系统将释放 locations 和 values 以及 X。但是，真正的问题是 X 占用的内存是否与 locations 和 values OR X 占用的内存相同是单独分配的内存，在这种情况下我需要释放 locations 和 values。

Answer 1

您正在使用的构造函数 (SpMat_meat.hpp:231)

template<typename T1, typename T2>
inline SpMat(const Base<uword,T1>& locations, const Base<eT,T2>&  values, const bool sort_locations = true);

用values 中值的拷贝 填充稀疏矩阵。

我知道您担心内存不足，如果您将 loc、val 和 X 分开，您基本上有两个相同数据的拷贝，占用的内存是实际需要的两倍(这确实是您的代码片段中发生的情况)，因此我将尝试着重解决这个问题并为您提供一些选择:

1) 如果您可以在短时间内保留两个数据拷贝，最简单的解决方案是动态分配loc 和val 并在 X

初始化后立即删除它们

int main(int argc, char**argv){
    umat* ploc;
    vec* pval;
    load_data(*ploc,*pval); 
    // at this point we have one copy of the data
    sp_mat X(*ploc,*pval); 
    // at this point we have two copies of the data
    delete ploc;
    delete pval;
    // at this point we have one copy of the data

    return 0;
}

当然你可以使用安全指针而不是 C 风格的指针，但你明白了。

2) 如果你绝对不想在任何时候拥有两份数据，我建议你修改你的load_data例程，一个一个地顺序加载值并直接插入它们进入 X

void load_data(umat &locations, vec& values, sp_mat& X){
 // fill up locations and values one by one into X
}

其他选项是 i) 使用移动语义直接移动 val 中的值到 X 或 ii) 直接使用内存分配给val作为X的内存，类似于矩阵的高级构造函数

Mat(eT* aux_mem, const uword aux_n_rows, const uword aux_n_cols, const bool copy_aux_mem = true, const bool strict = false)

然而，这两个选项都需要在 Armadillo 库级别进行修改，因为稀疏矩阵还没有提供这样的功能(目前只有一个普通的移动构造函数)。不过，向开发人员请求这些功能是个好主意!