本章将帮助您了解MXNet系统体系结构。让我们开始学习MXNet模块。

MXNet模块

下图是MXNet系统体系结构，它显示了MXNet模块的主要模块和组件以及它们之间的交互。

在上图中-

• 蓝色框中的模块是“面向用户的模块” 。

• 绿色框中的模块是系统模块。

• 实心箭头表示高度依赖，即严重依赖于界面。

• 虚线箭头表示光依赖性，即，为方便起见和接口一致性使用了数据结构。实际上，可以用替代方法代替它。

让我们讨论更多有关面向用户和系统模块的信息。

面向用户的模块

面向用户的模块如下-

• NDArray-它为Apache MXNet提供了灵活的命令式程序。它们是动态和异步的n维数组。

• KVStore-它充当有效参数同步的接口。在KVStore中，KV代表键值。因此，它是键值存储界面。

• 数据加载(IO) -此面向用户的模块用于高效的分布式数据加载和扩充。

• 符号执行-它是静态符号图执行器。它提供了有效的符号图执行和优化。

• 符号构建-该面向用户的模块为用户提供了一种构建计算图(即网络配置)的方法。

系统模组

系统模块如下-

• 存储分配器-顾名思义，此系统模块可在主机(即CPU)和不同设备(即GPU)上有效地分配和回收内存块。

• 运行时依赖引擎-运行时依赖引擎模块根据其读/写依赖关系调度并执行操作。

• 资源管理器-资源管理器(RM)系统模块管理全局资源，例如随机数生成器和时间空间。

• 运算符-运算符系统模块由所有定义静态正向和梯度计算(即反向传播)的运算符组成。