矩阵相乘的算法
很久没写blog了,感觉人都快变的抑郁了,换工作之后各种揪心,说好了是做Android的,结果让我搞各种算法,也罢,权当学习了一点知识吧。
今天说说矩阵相乘的算法,计算算法很简单,就是3个for循环。
首先还是说下矩阵相乘的概念,其实大学的时候线性代数中应该有讲到,不过到现在估计都还给老师了。
废话不多说,矩阵,其实就是一个二维数组,横竖排列的,比如int[5][6],就是一个矩阵,表示有5行6列。
只有当矩阵A的列数与矩阵B的行数相等时A×B才有意义。一个m×n的矩阵a(m,n)左乘一个n×p的矩阵b(n,p),会得到一个m×p的矩阵c(m,p)。左乘:又称前乘,就是乘在左边(即乘号前),比如说,A左乘E即AE。
在计算机中,一个矩阵实际上就是一个二维数组。一个m行n列的矩阵与一个n行p列的矩阵可以相乘,得到的结果是一个m行p列的矩阵,其中的第i行第j列位置上的数为第一个矩阵第i行上的n个数与第二个矩阵第j列上的n个数对应相乘后所得的n个乘积之和。比如,下面的算式表示一个2行2列的矩阵乘以2行3列的矩阵,其结果是一个2行3列的矩阵。
算法:
1 //矩阵相乘 2 public static float[][] Mul(float[][] a, float[][] b) { 3 //确保矩阵a的列数和b的行数相等 4 if(a[0].length != b.length) { 5 return null; 6 } 7 //用来存放结果的矩阵,axb的结果为a的行数和b的列数 8 float[][] result = new float[a.length][b[0].length]; 9 //对a的每行进行遍历 10 for(int i=0; i<a.length; i++) { 11 //对b的每列进行遍历 12 for(int j=0;j<b[0].length; j++) { 13 //c为每一个点的值 14 float c = 0; 15 //第i行j列的值为a的第i行上的n个数和b的第j列上的n个数对应相乘之和,其中n为a的列数,也是b的行数,a的列数和b的行数相等 16 for(int k=0; k<a[0].length; k++) { 17 c += (a[i][k]*b[k][j]); 18 } 19 result[i][j] = c; 20 } 21 } 22 return result; 23 }
代码注释的很清楚了,主要是抓住定义,3个for循环。如果你的二维数组不是float类型,可以相应的更改,记得将c和返回值一并更改。
就到这里吧。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· AI与.NET技术实操系列:向量存储与相似性搜索在 .NET 中的实现
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 分享 3 个 .NET 开源的文件压缩处理库,助力快速实现文件压缩解压功能!
· Ollama——大语言模型本地部署的极速利器
· [AI/GPT/综述] AI Agent的设计模式综述