二维矩阵实现矩阵相乘

最新推荐文章于 2025-07-15 11:54:32 发布

原创最新推荐文章于 2025-07-15 11:54:32 发布 · 1.6k 阅读

2 ·

CC 4.0 BY-SA版权

CUDA 专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍了如何使用Python的NumPy库高效地实现两个矩阵的相乘操作，包括基本概念、步骤和实例演示。通过理解矩阵乘法规则和NumPy的便捷函数，读者可以轻松掌握这一重要的数值计算技能。

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

   #include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
using namespace std;
#define M 1024
#define N 1024
#define P 512
//void mulMatri(int A[M][P],int B[P][N],int C[M][N],int m,int n,int p);
inline void mulMatri(float** A,float** B,float** C,int m,int n,int p);
int main()
{
    int i,j;
    float** A,**B,**C;
    A=(float**)malloc(M*sizeof(float *));
    B=(float**)malloc(P*sizeof(float *));
    C=(float**)malloc(M*sizeof(float *));
    for(i=0;i<M;i++)
        A[i]=(float*)malloc(P*sizeof(float));
    for(i=0;i<P;i++)
        B[i]=(float*)malloc(N*sizeof(float));
    for(i=0;i<M;i++)
        C[i]=(float*)malloc(N*sizeof(float));
    //int A[M][P],B[P][N],C[M][N];
    for(i=0;i<M;i++)
       for(j=0;j<P;j++)
         A[i][j]=rand()%10;
    for(i=0;i<P;i++)
       for(j=0;j<N;j++)
          B[i][j]=rand()%10;
    clock_t start,finish;
    start=clock();
    mulMatri( A,B,C,M,N,P);
    finish=clock();
    printf("%f ms\n",(float)(finish-start));
    system("pause");
    for(i=0;i<M;i++){
       for(j=0;j<P;j++)
          cout<<A[i][j]<<" ";
       cout<<endl;
    }
    for(i=0;i<P;i++){
       for(j=0;j<N;j++)
          cout<<B[i][j]<<" ";
       cout<<endl;
    }
    for(i=0;i<M;i++){
       for(j=0;j<N;j++)
          cout<<C[i][j]<<" ";
       cout<<endl;
    }
    for(i=0;i<M;i++)
        free(A[i]);
    for(i=0;i<P;i++)
        free(B[i]);
    for(i=0;i<M;i++)
        free(C[i]);
    free(A);
    free(B);
    free(C);
    return 0;
}
//void mulMatri(int A[M][P],int B[P][N],int C[M][N],int m,int n,int p)
inline void mulMatri(float** A,float** B,float** C,int m,int n,int p)
{
   int i,j,k;
   for(i=0;i<m;i++)
     for(j=0;j<n;j++){
         C[i][j]=0;
         for(k=0;k<p;k++)
            C[i][j]+=A[i][k]*B[k][j];
     }
}

来自为知笔记(Wiz)