[CUDA]CUDA编程实战四——矩阵乘法

矩阵乘法是最常见的操作,现代神经网络的基础便是矩阵乘法。 一个N*M的矩阵,乘以一个M*P的矩阵,得到N*P的矩阵,矩阵乘法即为将每一行与被乘矩阵对应列进行乘加,最后将所有结果进行汇总。 CPU版本 根据以上矩阵乘法的描述,便可以很快地实现矩阵乘法,三层循环,最内层循环做向量的乘加,最外的两层则做输
posted @ 2021-06-12 13:46  wildkid1024  阅读(4151)  评论(0编辑  收藏  举报