许多初学者觉得深度学习框架抽象,虽然调用了几个函数/方法,计算了几个数学难题,但始终不能理解这些框架的全貌。
为了更好地认识深度学习框架,也为了给一些想要自己亲手搭建深度学习框架的朋友提供一些基础性的指导,日前来自苏黎世联邦理工学院计算机科学系的硕士研究生Gokula Krishnan Santhanam在博客上撰文,概括了大部分深度学习框架都会包含的五大核心组件,为我们详细剖析了深度学习框架一般性的内部组织结构。以下由雷锋网(公众号:雷锋网)编译。
Gokula Krishnan Santhanam认为,大部分深度学习框架都包含以下五个核心组件:
>
- 张量(Tensor)
- 基于张量的各种操作
- 计算图(Computation Graph)
- 自动微分(Automatic Differentiation)工具
BLAS、cuBLAS、cuDNN等拓展包