mat_mult.cpp

来自「本程序将独立分量分析技术和数字水印技术有机地结合在一起」· C++ 代码 · 共 33 行

#include <stdio.h>
/*mat_mult-矩阵相乘子程序
 *把n×p的矩阵A乘于q×r的矩阵B, 即C=A*B，得到n×r的矩阵C 
 */
void mat_mult (float *A, int n, int p, float *B, int q, int r, float *C)
{   

	int i,j,k;
    if (p != q) 
	{
		printf("Error, matrices not suitable\nfor multiplication");
	}
	else 
	{
	for(i=0;i<n;i++)
        for(j=0;j<r;j++)
        {       
			    C[i*r+j]=0;
			    k=0;    

                while(k<p)
			{  
				
				C[i*r+j]+=A[i*p+k]*B[k*r+j];
                k++;
            }
        }

		
	}
	
}