DL

矩阵相乘效率比较

矩阵相乘

Posted by Ezra on May 26, 2018

矩阵相乘效率比较 2018年04月12日 10:07:39 阅读数:61 牛客网上的一道题 深度学习是当前很热门的机器学习算法,在深度学习中,涉及到大量的矩阵相乘,现在需要计算三个稠密矩阵A,B,C的乘积ABC,假设三个矩阵的尺寸分别为mn,np,pq,且m<n<p<q,以下计算顺序效率最高的是() A. ABC B. (AC)B C. A(BC) D. 所有效率都相同 解答:考查矩阵相乘的效率问题,即需要计算的乘法和加法的次数之和。当m,n,p,q较大时,加法忽略不计。任意两个大小分别为ab,bc的矩阵相乘,需要乘法的次数为abc。 因此,A选项的乘法计算次数,mnp+mpq;B选项不满足矩阵相乘规则,舍弃;C选项的乘法计算次数,npq+mnq 其中,m<n<p<q。所以mnp<mnq,mpq<np*q,选A