Caffe源码剖析3-InnerProduct层

首先引入对接口caffe_gemm和caffe_gemv的介绍

template<>
void caffe_cpu_gemm<float>(const CBLAS_TRANSPOSE TransA,
    const CBLAS_TRANSPOSE TransB, const int M, const int N, const int K,
    const float alpha, const float* A, const float* B, const float beta,
    float* C) {
  int lda = (TransA == CblasNoTrans) ? K : M;
  int ldb = (TransB == CblasNoTrans) ? N : K;
  cblas_sgemm(CblasRowMajor, TransA, TransB, M, N, K, alpha, A, lda, B,
      ldb, beta, C, N);
}

其中TransA表示是否对A进行转置，TransB表示是否对B进行转置。

const int M，矩阵A的行，矩阵C的行

const int N，矩阵B的列，矩阵C的列

const int K，矩阵A的列，矩阵B的行

换言之：A: M*K B: K*N C: M*N，这个还是不直观，我画了了一个图，相信你永远也不会忘记了。