3. 计算图(DONE)¶
为了高效地训练一个复杂神经网络,框架需要解决诸多问题, 例如:如何实现自动求导,如何利用编译期分析对神经网络计算进行化简、合并、变换,如何规划基本计算单元在加速器上的执行,如何将基本处理单元派发(dispatch)到特定的高效后端实现,如何进行内存预分配和管理等。用统一的方式解决这些问题都驱使着框架设计者思考如何为各类神经网络计算提供统一的描述,从而使得在运行神经网络计算之前,能够对整个计算过程尽可能进行推断,在编译期自动为用户程序补全反向计算,规划执行,最大程度地降低运行时开销。
目前主流的深度学习框架都选择使用计算图来抽象神经网络计算,《计算图》实际上,AI框架主要的职责是把深度学习的表达转换称为计算机能够识别的计算图,计算图作为AI框架中核心的数据结构,贯穿AI框架的大部分整个生命周期,于是计算图对于AI框架的前端核心技术就显得尤为重要。
希望这个系列能够给大家、朋友们带来一些些帮助,也希望自己能够继续坚持完成所有内容哈!
内容大纲
PPT和字幕需要到 Github 下载,网页课程版链接会失效哦~建议优先下载 PDF 版本,PPT 版本会因为字体缺失等原因导致版本很丑哦~
小节 |
链接 |
|---|---|
01 基本介绍 |
|
02 什么是计算图 |
|
03 与自动微分关系 |
|
04 图优化与图执行调度 |
|
05 计算图控制流实现 |
|
06 计算图实现动静统一 |
|
07 计算图的挑战与未来 |