c++ arrayfire库 多维数组矩阵相乘

void af_matmul_test2() {
    int m = 2, k = 2, q = 3, n = 2, p = 4;
    af::array A = af::randu(m, k, n, p); //m*k的随机矩阵，第3维是n,第4维是p
    af::array B = af::randu(k, q, n, p); //k*q的随机矩阵，第3维是n,第4维是p
    af_print(A);
    af_print(B);
    af::array AB = af::matmul(A, B); //m*q的随机矩阵，第3维是n,第4维是p
    af_print(AB);
    std::cout << "leave " << __func__ << std::endl;

}

2个多维数组矩阵相乘，一个数组维度是(m, k, n, p), 另一个数组维度是(k, q, n, p),第3维n和第4维p是相同的，然后第1维和第2维满足矩阵相乘要求，即m行k列乘以k行q列