C/C++ 中 mdspan 介绍与使用示例

#include <mdspan>

#include <mdspan>
#include <vector>
#include <iostream>
int main() {
    std::vector<int> data = {1, 2, 3, 4, 5, 6};
    // 创建一个 2x3 的 mdspan 视图
    std::mdspan mat{data.data(), 2, 3};
    // 访问元素
    std::cout << mat(1, 2) << '\n';  // 输出 6
}

#include <mdspan>
int main() {
    int data[6] = {1, 2, 3, 4, 5, 6};
    // 使用行优先布局 (C风格)
    std::mdspan mat_c{data, std::extents{2, 3}, std::layout_right{}};
    // 使用列优先布局 (Fortran风格)
    std::mdspan mat_f{data, std::extents{2, 3}, std::layout_left{}};
    // 自定义跨步布局
    std::mdspan mat_stride{data, std::extents{2, 3}, std::layout_stride{std::array{1, 2}}};
}

#include <mdspan>
int main() {
    int data[24];
    // 完全静态维度
    std::mdspan<int, std::extents<2, 3, 4>> static_span{data};
    // 混合静态和动态维度
    std::mdspan<int, std::extents<std::dynamic_extent, 3, std::dynamic_extent>> mixed_span{data, 2, 4};
    // 完全动态维度
    std::mdspan<int, std::extents<std::dynamic_extent, std::dynamic_extent>> dynamic_span{data, 2, 3};
}

#include <mdspan>
#include <vector>
#include <iostream>
template <typename T, typename ExtsA, typename LayA, typename AccA,
          typename ExtsB, typename LayB, typename AccB,
          typename ExtsC, typename LayC, typename AccC>
void matrix_multiply(
    std::mdspan<T, ExtsA, LayA, AccA> a,
    std::mdspan<T, ExtsB, LayB, AccB> b,
    std::mdspan<T, ExtsC, LayC, AccC> c) {
    for (size_t i = 0; i < c.extent(0); ++i) {
        for (size_t j = 0; j < c.extent(1); ++j) {
            c(i, j) = 0;
            for (size_t k = 0; k < a.extent(1); ++k) {
                c(i, j) += a(i, k) * b(k, j);
            }
        }
    }
}
int main() {
    std::vector<int> a_data(2*3, 1); // 2x3 矩阵，全1
    std::vector<int> b_data(3*4, 2); // 3x4 矩阵，全2
    std::vector<int> c_data(2*4);    // 2x4 结果矩阵
    auto a = std::mdspan(a_data.data(), 2, 3);
    auto b = std::mdspan(b_data.data(), 3, 4);
    auto c = std::mdspan(c_data.data(), 2, 4);
    matrix_multiply(a, b, c);
    // 输出结果
    for (size_t i = 0; i < c.extent(0); ++i) {
        for (size_t j = 0; j < c.extent(1); ++j) {
            std::cout << c(i, j) << ' ';
        }
        std::cout << '\n';
    }
}