BLAS dgemm矩阵乘法的用法小注

BLAS dgemm矩阵乘法的用法小注

Posted on 04月 5, 2010 by xuanmanstein

http://xuanmanstein.wordpress.com/2010/04/05/blas-dgemm%E7%9F%A9%E9%98%B5%E4%B9%98%E6%B3%95%E7%9A%84%E7%94%A8%E6%B3%95%E5%B0%8F%E6%B3%A8/

矩阵乘法C = A*B

A:M*k  B:k*N C:M*N

dgemm(&transa,&transb,&m,&n,&k,&alpha,A[0],&lda,B[0],&ldb,&beta,C[0],&ldc);

这个算法看得晕头转向，用得也是跌跌撞撞。注记如下：

1．参数

其中，m n 就是C的最后尺寸（m行，n列），不论AB 是否转置。

k是A B 两阵相等的维数。因A B 转置而不同

后面那三个就是3个阵 每个阵的最高的维数

m = M;

n = N;

k = ?;

lda = max(m,k);

ldb = max(n,k);

ldc = max(m,n);

 

2. 保存结果C一定要单独定义一个变量，即使维数和A B中某阵相同，也不能图“节约内存”，用 AB之一保存结果。