Skip to main content
 Web开发网 » 站长学院 » 浏览器插件

MXNet快速入门之NDArray详细介绍

2021年11月04日6190百度已收录

前言由于最近需要用到mxnet,所以准备结合mxnet官网教程,写一系列mxnet的快速入门教程。通过一系列文章你将能够快速掌握mxnet的核心NDArray和Gluron以及灵活利用mxnet来构建神经网络,mxnet中的NDArray数据结构相当于TensorFlow中的tensor,其结构有点类似于numpy的array,通过mxnet的Gluron可以快速来构建神经网络。接下来的这篇文章,将详细介绍一下NDArray。

MXNet是亚马逊(Amazon)公司的一个深度学习框架,为多GPU配置提供了良好的设置,多个GPU之间的切换只需要一句代码。同时使用MXNet来构建一个神经网络架构也是非常简单,在后面的教程中会详细介绍。

mxnet官网教程:

MXNet快速入门之NDArray详细介绍  Mxnet教程 第1张

MXNet安装教程mxnet官网安装教程:

根据自己的需要安装GPU或CPU版本,具体安装请参考MXNet安装教程根据自己的系统以及需要安装的版本等信息进行选择,然后根据生成的命令安装,该系列教程主要包括以下内容:

MXNet中的核心数据类型NDArray介绍如何使用MXNet来构建一个神经网络利用autograd来计算链式求导训练神经网络利用预训练模型来进行预测如何使用GPU加速运算NDArray

通过元组创建NDArrayMXNet快速入门之NDArray详细介绍  Mxnet教程 第2张

利用内置函数创建矩阵MXNet快速入门之NDArray详细介绍  Mxnet教程 第3张

查看NDArray的相关信息MXNet快速入门之NDArray详细介绍  Mxnet教程 第4张

NDArray的乘法MXNet快速入门之NDArray详细介绍  Mxnet教程 第5张

注意:使用*的乘法,即一对一的乘法,不是矩阵的乘法。所以x的shape需要与z的shape相同或者z的shape为1,当z的shape为1时,在计算乘法的时候z的shape会自动扩展与x相同这种操作被称为广播。

将矩阵中的元素求e(自然底数)的次方MXNet快速入门之NDArray详细介绍  Mxnet教程 第6张

矩阵的乘法MXNet快速入门之NDArray详细介绍  Mxnet教程 第7张

注意:在计算矩阵的乘法时,前一个矩阵的列数必须要等于后一个矩阵的行数。因为要满足矩阵点乘公式,矩阵之间的乘法计算如下:

MXNet快速入门之NDArray详细介绍  Mxnet教程 第8张

矩阵的转置MXNet快速入门之NDArray详细介绍  Mxnet教程 第9张

NDArray的切片MXNet快速入门之NDArray详细介绍  Mxnet教程 第10张

将NDArray转为numpy的arrayMXNet快速入门之NDArray详细介绍  Mxnet教程 第11张

最近开通了微信公众号,在微信公众号上,分享了一些资料,搜索"Python机器学习之路"关注我即可获取,欢迎来找我玩。

评论列表暂无评论
发表评论
微信