2021年6月12日
[CUDA]CUDA编程实战四——矩阵乘法
摘要： 矩阵乘法是最常见的操作，现代神经网络的基础便是矩阵乘法。 一个N*M的矩阵，乘以一个M*P的矩阵，得到N*P的矩阵，矩阵乘法即为将每一行与被乘矩阵对应列进行乘加，最后将所有结果进行汇总。 CPU版本 根据以上矩阵乘法的描述，便可以很快地实现矩阵乘法，三层循环，最内层循环做向量的乘加，最外的两层则做输 阅读全文
posted @ 2021-06-12 13:46 wildkid1024 阅读(4436) 评论(0) 推荐(0) 编辑