映射特征矩阵Ref对象与块不一致

Mapping Eigen Matrix Ref objects not consistent with blocks

本文关键字:不一致 对象 Ref 特征 映射      更新时间:2023-10-16

快速上下文,我正在使用另一个C++库,该库具有期望正则或映射特征矩阵的函数。我想使用映射版本来避免复制时的内存开销。

也就是说,我正在尝试使用矩阵块。我知道使用返回Eigen::BlockEigen::Ref对象的block方法可以很容易地访问这些对象。下面我正在尝试使用Ref对象。我想Map Eigen::MatrixXd块。然而,我似乎无法跨列映射,只能映射列中的连续元素(我认为这是面向列的数据的结果)。您可以在下面看到输出的差异。

有什么办法让我把Map变成Eigen::MatrixXd的一块吗?

#include <iostream>
#include <Eigen/Core>
int main()
{
    Eigen::MatrixXd A(3,3);
    A(0,0) = 1.0;
    A(0,1) = 2.0;
    A(0,2) = 3.0;
    A(1,0) = 4.0;
    A(1,1) = 5.0;
    A(1,2) = 6.0;
    A(2,0) = 7.0;
    A(2,1) = 8.0;
    A(2,2) = 9.0;
    std::cout << "source" << std::endl;
    std::cout << A << std::endl;
    Eigen::Ref<Eigen::MatrixXd> block = A.block(1,1,1,2);
    std::cout << "block" << std::endl;
    std::cout << block << std::endl;
    Eigen::Map<Eigen::MatrixXd> map(block.data(), block.rows(), block.cols());
    std::cout << "map" << std::endl;
    std::cout << map << std::endl;
}

输出:

source
1 2 3
4 5 6
7 8 9
block
5 6
map
5 8

除非另有规定,否则Eigen::Map采用单位步幅。Ref对象的问题在于步幅不是1。您可以指定步幅(在这种情况下是外侧)如下:

Eigen::Map<Eigen::MatrixXd, 0, Eigen::OuterStride<> >
       map2(block.data(), block.rows(), block.cols(), Eigen::OuterStride<>(3));
std::cout << "map2" << std::endl;
std::cout << map2 << std::endl;

更好的是,您可以使用Ref对象的外部步幅为地图指定它:

Eigen::Map<Eigen::MatrixXd, 0, Eigen::OuterStride<> >
   map2(block.data(), block.rows(), block.cols(), Eigen::OuterStride<>(block.outerStride()));