You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.10.03 08:15:03