时间戳

首页 > 解决方案 > 如何在 Eigen 3.3.4 中的稀疏块上实例化 InnerIterator？

block - 如何在 Eigen 3.3.4 中的稀疏块上实例化 InnerIterator？

问题描述

我有一段代码在 Eigen 3.2 中运行良好，但在 Eigen 3.3.4 中不再有效。这是代码：

// Temporary Eigen blocks
Eigen::Block<const Eigen::SparseMatrix<double> > 
tmpAPotentialBlock(A.block(startPotential, startPotential, sizePotential,sizePotential)), 
tmpAFlowBlock(A.block(startFlow, startPotential, sizeFlow, sizePotential));

for (Eigen::SparseMatrix<double>::Index k=0; k<sizePotential; ++k) {
  // Iterator to the first term of the column k of the potential block and the flow block.
  Eigen::Block<const Eigen::SparseMatrix<double> >::InnerIterator itAPotential(tmpAPotentialBlock,k),
                                                                itAFlow(tmpAFlowBlock,k);
  ...
}

基本上问题是InnerIterator不再为块或至少稀疏块定义。

我知道您现在需要使用 anevaluator来定义它。有谁知道新语法是什么？

标签： blocksparse-matrixeigeneigen3

解决方案

你需要写：

Eigen::InnerIterator<SpBlock> it(tmp,k)

这是一个独立的 C++11示例：

using SpMat = Eigen::SparseMatrix<double>;
using SpBlock = Eigen::Block<const SpMat>;
SpMat A;
Index i, s;
SpBlock tmp(A, i, i, s, s);

for (Eigen::Index k=0; k<s; ++k) {
    Eigen::InnerIterator<SpBlock> it(tmp,k);
    /* ... */
}

在 C++17 中可以变得更漂亮：

Eigen::SparseMatrix<double> A;
Index i, s;
auto tmp = A.block(i, i, s, s);

for (Eigen::Index k=0; k<s; ++k) {
    Eigen::InnerIterator it(tmp,k);
    /* ... */
}

推荐阅读