caffe的math_functions全解
caffe的math_functions全解gemmaxpy持续更新中,请期待
gemm
gemm是实现两个矩阵相乘,实现有caffe_cpu_gemm和caffe_gpu_gemm,如有两个矩阵A和B相乘得到矩阵C,调用方法如下:
CPU版:
caffe_cpu_gemm(CblasNoTrans, CblasNoTrans, m, n, k, alpha,
A.cpu_data(), B....
原创
2019-08-24 12:04:12 ·
332 阅读 ·
0 评论