c++ arrayfire库矩阵分块,上下分块对调

rayyy9

于 2024-10-02 21:17:01 发布

阅读量88

点赞数 1

分类专栏： arrayfire 文章标签： c++ 笔记

本文链接：https://blog.csdn.net/reyyy/article/details/142685353

版权

arrayfire 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

上下分块对调

void gfor_test5() {

    int m = 4, n = 4;
    af::array A = randu(m, n, 2, 2);
    af_print(A);

    af::array A1 = A(af::seq(2), af::span, af::span, af::span);  //上面一半分块
    af::array A2 = A(af::seq(2, af::end), af::span, af::span, af::span); //下面一半分块
    af_print(A1);
    af_print(A2);

    //上面一半分块和下面一半分块对调
    af::array b = randu(m, n, 2, 2);
    b(af::seq(2), af::span, af::span, af::span) = A2;
    b(af::seq(2, af::end), af::span, af::span, af::span) = A1;
    af_print(b);

}