上QQ阅读APP看书,第一时间看更新
1.4.2 NumPy
NumPy(Numerical Python)是Python语言中用于科学计算的非常基础和重要的库,支持大量的数组和矩阵运算。NumPy中最常用的数据类型是array,array翻译过来就是数组的意思,在NumPy中array可以是多维的,比如0维的array就是标量,1维的array就是向量,2维的array就是矩阵等。
为什么要了解NumPy呢?因为大多数深度学习框架的基础数据结构都参考了NumPy中的array,比如MXNet框架中的NDArray、TensorFlow和PyTorch框架中的Tensor等。那么既然有NumPy array,为什么不直接在框架中使用这种数据结构呢?主要原因在于NumPy array只能在CPU上运行,不能在GPU上运行,因此在MXNet中就引入了NDArray,NDArray的大部分用法与NumPy array相似,最大的不同点在于NDArray可以在GPU上运行。因此,了解和熟悉NumPy的相关知识对于后续学习MXNet的NDArray接口以及其他代码实现都有一定的帮助。