C++如何实现矩阵乘法运算

2026-06-03编程代码286088

std::vector实现二维矩阵乘法需三层循环:i遍历行、j遍历列、k做点积,要求A列数等于B行数,结果C为m×p且预分配,k范围是[0,n)。

std::vector 实现二维矩阵并手写乘法循环

直接用 std::vector> 表示矩阵最直观,也便于动态尺寸。但要注意:必须保证每行长度一致,否则乘法会越界或逻辑错误。

核心是三层嵌套循环:外层遍历结果矩阵的行(i),中层遍历列(j),内层做点积(k)。关键约束是:左矩阵列数必须等于右矩阵行数,否则乘法无定义。

  • 检查维度:若 Am×nBn×p,则结果 C 必须是 m×p,且初始化为全 0
  • 内层循环索引 k 的范围是 [0, n),不是 [0, m)[0, p)——这是最常见的下标错位
  • 避免重复构造临时 vector:不要在循环里写 std::vector(p, 0.0) 创建新行,应预先分配好 C 的结构
std::vector<std::vector> matmul(
    const std::vector<std::vector>& A,
    const std::vector<std::vector>& B) {
    size_t m = A.size(), n = A[0].size(), p = B[0].size();
    std::vector<std::vector> C(m, std::vector(p, 0.0));
    for (size_t i = 0; i < m; ++i)
        for (size_t j = 0; j < p; ++j)
            for (size_t k = 0; k < n; ++k)
                C[i][j] += A[i][k] * B[k][j];
    return C;
}

用一维 std::vector 存储提升缓存局部性

二维 vector 在内存中不连续,频繁跨行访问会引发大量 cache miss。换成一维存储(行优先)后,内层循环能顺序读取内存,实测在中等规模(如 1000×1000)下可提速 2–3 倍。

索引公式必须严格:元素 A[i][j] 对应一维下标 i * cols + jB[k][j] 对应 k * p + j(注意不是 j * n + k)。

  • 传参时需额外携带行列信息:matmul(const std::vector& A, size_t Am, size_t An, const std::vector& B, size_t Bn, size_t Bp)
  • 内层循环仍是对 k,但访问模式变成 A[i*An + k]B[k*Bp + j] —— 错写成 B[j*Bn + k] 会导致结果全错
  • 如果后续要和 BLAS 库对接,一维布局是强制要求,提前统一格式可减少转换开销

遇到 std::out_of_range 或结果全 0 怎么快速定位

这两个现象几乎都源于维度校验缺失或索引越界。不要直接看乘法循环,先检查三处:

C知道

CSDN推出的一款AI技术问答工具

下载

  • 输入矩阵是否为空:A.empty() || A[0].empty() || B.empty() || B[0].empty()
  • 是否满足乘法前提:A[0].size() != B.size() 时应抛异常或返回空矩阵,而不是硬算
  • 内层循环上限写错:比如把 k 误写成 k ,会导致访问 A[i][k] 越界(k 超出列数)

调试时可在循环内加一句 assert(k ,比靠 operator[] 抛异常更快暴露问题。

什么时候该换 BLAS 而不是手写

手写适合教学、小规模()或特殊需求(如自定义数值类型、稀疏逻辑)。一旦矩阵超过 500×500,OpenBLAS 或 Intel MKL 的优化版本通常快一个数量级以上,且自动适配多核与 SIMD。

调用 cblas_dgemm 时最容易漏的是:矩阵按列优先(Fortran 风格)还是行优先(C 风格)——C 接口默认列优先,若你用行优先数据,必须把参数 layout 设为 CblasRowMajor,否则结果完全错误。

还有个隐形坑:BLAS 要求内存对齐(尤其是 AVX 指令),如果 vector 分配的地址未 32 字节对齐,某些实现会降级到 SSE 或崩溃。生产环境建议用 aligned_alloc 或 Eigen 等封装好的库来规避。

本文地址:https://www.ufcn.cn/article/2412367.html

如非特殊说明,本站内容均来自于网友自主分享,概不代表本站观点,如有任何问题我们都将在收到反馈后的第一时间进行处理!