我知道Caffe使用GEneral矩阵进行矩阵乘法,它是用于执行卷积运算的基本线性代数子程序库的一部分。其中卷积被转换为矩阵乘法运算。我已经提到了下面的条款。https://petewarden.com/2015/04/20/why-gemm-is-at-the-heart-of-deep-learning/
我想了解其他深层次的学习框架,如Theano,Tensorflow,Py火炬是如何执行卷积操作的。他们是否在后端使用类似的库。可能会有一些关于这个主题的文章。如果有人能指给我看或者能用答案解释的话。
发布于 2018-08-13 09:16:16
这些框架基于Cuda,这是GPU的并行计算工具。尤其是深度学习框架所使用的是cuDNN。
您可以找到用户指南:https://docs.nvidia.com/deeplearning/sdk/pdf/cuDNN-Developer-Guide.pdf
所有的操作都被描述,实际上GEMM算法被用于卷积运算。
https://datascience.stackexchange.com/questions/36841
复制相似问题