caffe源码中关于矩阵运算的一些函数

本文深入探讨了Caffe框架中用于矩阵运算的函数,这些函数主要在`math_function.cpp`和`math_function.hpp`文件中定义。这些函数实际上是CBLAS库的简化版本,依赖于CBLAS的接口进行操作。通过分析`math_function.cpp`的代码,我们可以理解如何使用CBLAS函数处理矩阵,包括矩阵乘法、标量乘法和加法等操作。关键在于理解CBLAS函数如何通过指针和参数描述矩阵的形状和存储布局。在C++中,矩阵数据按行存储,与Fortran的列存储方式不同。学习Caffe矩阵运算函数的方法是首先找到调用的CBLAS函数,然后查阅CBLAS手册获取详细信息。
摘要由CSDN通过智能技术生成

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 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值