如何在opencv中使用GPU将矩阵与3维相乘

时间:2015-10-22 11:46:40

标签: c++ opencv matrix gpu

opencv文档说GpuMat只适用于二维矩阵,但是我在一个二维上抽象出三维矩阵,我使用一个循环来使用gpu::gemm

来乘以这两个矩阵。
for (int k = 0; k <C, k ++) {
    gpu::gemm (matrix_a.colRange (Range (k * M (k + 1) * M)),
    matrix_b.rowRange (Range (k * M (k + 1) * M)),
    N [k + 1], matrix_c, 1, matrix_c);
}

其中M用于移动另一个矩阵。 matrix_a matrix_bmatrix_cgpu::GpuMat

正如你所看到的,循环是串行的,我想做的是并行化整个操作,有人有什么建议吗?

0 个答案:

没有答案