Eigen C++如何用Map实现无拷贝拼接奇偶行来自不同矩阵的新矩阵
Eigen无拷贝交叉合并矩阵实现方案
首先修正你示例中的笔误:VectorXd A(a,b)是错误用法,VectorXd为一维动态向量类型,存储二维矩阵需使用MatrixXd,以下实现基于MatrixXd类型展开。
核心思路
由于A、B是两个独立存储的内存块,无法直接通过单个Eigen::Map构造连续的合并矩阵(Map仅支持指向单块连续内存),因此采用Eigen惰性表达式模板实现无拷贝视图:通过惰性纵向拼接表达式+索引重排,构造出的合并矩阵视图不会分配额外内存,所有读写操作直接映射到原始A、B的内存地址,完全避免内存翻倍问题。
完整实现代码
#include <Eigen/Core> #include <Eigen/Dense> int main() { // 模拟输入:A为顶点矩阵(3行3列),B为法向量矩阵(2行3列),列数b=d=3符合预设条件 Eigen::MatrixXd A(3, 3); A << 1,2,3, 7,8,9, 13,14,15; Eigen::MatrixXd B(2, 3); B << 4,5,6, 10,11,12; const int total_rows = A.rows() + B.rows(); const int cols = A.cols(); // 1. 构造惰性纵向拼接表达式,无任何内存拷贝 auto AB_stack = Eigen::vstack(A, B); // 2. 生成交叉行索引序列,仅生成索引数组无矩阵元素拷贝 Eigen::VectorXi row_indices(total_rows); int a_idx = 0, b_idx = A.rows(); for (int i = 0; i < total_rows; ++i) { if (i % 2 == 0) { // 第i+1行为奇数行,取A的对应行 row_indices(i) = a_idx++; } else { // 第i+1行为偶数行,取B的对应行 row_indices(i) = b_idx++; } } // 3. 构造无拷贝的合并矩阵视图C,所有操作直接映射到原始A、B内存 auto C = AB_stack(row_indices, Eigen::all); // 测试验证:修改C的元素会同步修改原始A/B的值 C(0, 0) = 99; // 等价于 A(0,0) = 99 C(1, 0) = 100; // 等价于 B(0,0) = 100 return 0; }
注意事项
- 该实现依赖Eigen 3.4及以上版本,低版本Eigen可自行实现惰性拼接逻辑替代
Eigen::vstack - 上述代码中的循环仅用于生成行索引,不会拷贝任何矩阵元素,符合无拷贝要求
C为Eigen表达式类型,而非MatrixXd实例,建议用auto自动推导类型;如果业务要求C必须是连续存储的MatrixXd类型,则无法避免内存拷贝(因为MatrixXd要求自身内存连续,而A、B是独立的两块内存),仅当A、B原始内存本身就是交叉排列时,才能通过Map构造无拷贝的MatrixXd视图
内容的提问来源于stack exchange,提问作者Kabu
相关产品推荐
相关产品推荐

