文章来源于网络收集而来,版权归原创者所有,如有侵权请及时联系!
分块矩阵
经典的矩阵乘法
问题:计算两个 n×n
的矩阵 A×B = C
的矩阵乘法的计算复杂度(大 O
)是多少?
你可以在 Codecademy 学习(或复习)大 O
。
它的样子是:
for i=1 to n
{read row i of A into fast memory}
for j=1 to n
{read col j of B into fast memory}
for k=1 to n
C[i,j] += A[i,k] x B[k,j]
{write C[i,j] back to slow memory}
问题:进行了多少次读写操作?
分块矩阵相乘
将 A
, B
, C
分成大小为 N/n × N/n
的 N×N
个块。
它的样子是:
for i=1 to N
for j=1 to N
for k=1 to N
{read block (i,k) of A}
{read block (k,j) of B}
block (i,j) of C += block of A times block of B
{write block (i,j) of C back to slow memory}
问题 1:这个的大 O
是什么?
问题 2:进行了多少次读写操作?
如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱,如果其他用户或者作者回复了您的评论,将不能在第一时间通知您!
发布评论