caffe中的一些矩阵运算函数,定义在文件caffe\src\caffe\util\math_function.cpp和文件caffe\include\caffe\util\math_function.hpp里
它们实际上是cblas的精简版,因此,math_function.hpp的头文件一般包含cblas相关文件。
当我们在某些文件中出现关于矩阵运算的函数的时候,首先在math_function.cpp里找到该函数名称,查看该函数调用的cblas函数名,然后,查看cblas手册,得到该函数的具体描述。
矩阵中的数据存储在一片连续的内存中,即它的数据结构实际上是vector。cblas中的函数通过指向这些内存首地址的指针来访问矩阵中的数据。为了让这些vector具有矩阵的特性,还要向cblas中的函数传递用来描述这些矩阵形状、尺寸等信息的参数,例如矩阵的行数、列数、是否转置等等。
下面是math_function.cpp文件中的一段段代码:
</pre><pre style="margin-top: 0px; margin-bottom: 0px;"><span style=" color:#808000;">template</span><>
void caffe_cpu_gemm<float>(const CBLAS_TRANSPOSE TransA,
const CBLAS_TRANSPOSE