C++实现矩阵乘

最新推荐文章于 2024-02-10 22:20:09 发布

Alphapeople

最新推荐文章于 2024-02-10 22:20:09 发布

阅读量452

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/weixin_38241876/article/details/110425523

版权

算法专栏收录该内容

5 篇文章 0 订阅

订阅专栏

#include<iostream>
using namespace std;
#define ROWS 2
#define COLS 3

void function(float *A,float *B,float *C)
{
	for(int i=0;i<ROWS;i++)
	{
		for(int j=0;j<ROWS;j++)
		{
			float sum = 0.0;
			for(int k=0;k<COLS;k++)
			{
				float a = A[i*COLS+k];
				float b = B[k*ROWS+j];
				sum += a*b;
			}
			C[i*ROWS+j] = sum;
			cout<<sum<<" ";
		}
		cout<<endl;
	}
}

int main()
{
	float *A,*B,*C;
	float s = 1.0;
	int totalsize = ROWS*COLS*sizeof(float);
	A = (float *)malloc(totalsize);
	B = (float *)malloc(totalsize);
	C = (float *)malloc(totalsize);
	//初始化矩阵
	for(int i=0;i<ROWS;i++)
	{
		for(int j=0;j<COLS;j++)
		{
			A[i*COLS+j] = s;
			B[j*ROWS+i] = s;
			s++;
		}
	}
	function(A,B,C);
	return 0;
}