c++ - 在不复制的情况下从 Eigen::SparseMatrix 中提取 block /ROI

Question

我想知道有没有什么好的方法可以从 Eigen::SparseMatrix 中提取 block /ROI？更准确地说，我要提取的是内部 vector 。

我想做的是这样的:

typedef Eigen::SparseMatrix<double,Eigen::RowMajor> SpMat;
// Prepare some sparse matrix
SpMat spmat;
// Extract lines from it
const SpMat& row_i = spmat.innerVector(i);
const SpMat& row_j = spmat.innerVector(j);
// Some calculation with row_i and row_j...

根据我的测试，row_i 和 row_j 的数据是从 spmat 中复制 (!!) 的。然而，显然，它是低效的。内部 vector 的数据(尤其是row_i.m_data.m_values & row_i.m_data.m_indices)是原始数据(spmat.m_data.m_values & spmat.m_data.m_indices 分别)，所以应该有更聪明的方法。

我也许能够实现新方法来执行此操作，但这需要我深入挖掘源代码。所以我不想。

如有帮助，不胜感激!提前致谢。

Answer 1

您可以使用 c++11 auto 关键字将 row_i 和 row_j 声明为真正的读写表达式，或者使用正确的类型:

const auto row_i = spmap.innerVector(i); // C++11 version
const SpMat::InnerVectorReturnType row_i = spmap.innerVector(i); // C++98 version

此外，并不是说默认情况下 SparseMatrix 存储在主要列中，因此“内部 vector ”是一个列。如果要引用行，则必须使用行优先存储布局:

typedef Eigen::SparseMatrix<double,RowMajor> SpMat;