首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >GEMM在Tensorflow,Theano,Py火炬中使用吗?

GEMM在Tensorflow,Theano,Py火炬中使用吗?
EN

Data Science用户
提问于 2018-08-12 19:31:04
回答 1查看 1.5K关注 0票数 1

我知道Caffe使用GEneral矩阵进行矩阵乘法,它是用于执行卷积运算的基本线性代数子程序库的一部分。其中卷积被转换为矩阵乘法运算。我已经提到了下面的条款。https://petewarden.com/2015/04/20/why-gemm-is-at-the-heart-of-deep-learning/

我想了解其他深层次的学习框架,如Theano,Tensorflow,Py火炬是如何执行卷积操作的。他们是否在后端使用类似的库。可能会有一些关于这个主题的文章。如果有人能指给我看或者能用答案解释的话。

EN

回答 1

Data Science用户

发布于 2018-08-13 09:16:16

这些框架基于Cuda,这是GPU的并行计算工具。尤其是深度学习框架所使用的是cuDNN。

您可以找到用户指南:https://docs.nvidia.com/deeplearning/sdk/pdf/cuDNN-Developer-Guide.pdf

所有的操作都被描述,实际上GEMM算法被用于卷积运算。

票数 0
EN
页面原文内容由Data Science提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://datascience.stackexchange.com/questions/36841

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档