caffe: math_functions

最新推荐文章于 2025-08-06 17:54:46 发布

原创最新推荐文章于 2025-08-06 17:54:46 发布 · 384 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#caffe

caffe 专栏收录该内容

4 篇文章

订阅专栏

本文详细介绍了 Caffe 框架中基于 CPU 的两种核心数学运算：gemv 和 gemm。gemv 主要用于实现向量与矩阵的乘法运算，而 gemm 则针对矩阵间的乘法运算进行了优化。这两种操作都支持转置功能，并能够灵活地调整输入输出矩阵的维度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1、

/*  A:m*n
 *  if CblasTrans
 *      B:m*1
 *      C:n*1
 *      C=alpha*A'*B+beta*C
 *  else
 *      B:n*1
 *      C:m*1
 *      C=alpha*A*B+beta*C
 *  end
 */ 
caffe_cpu_gemv<Dtype>(CblasTrans/CblasNoTrans,
        m, n, alpha, A, B, beta, C);

2、

/*  C:m*n
 *  op(A):m*k
 *  op(B):k*n
 *  C=alpha*op( A )*op( B ) + beta*C
 *  function X=op(X)
 *      if CblasTrans
 *          X=X'
 *      else
 *          X=X
 *      end
 */
caffe_cpu_gemm<Dtype>(CblasTrans/CblasNoTrans,
        CblasTrans/CblasNoTrans,
        m, n, k, alpha, A, B, beta, C);