动手学深度学习之预备知识（MXNET）

最新推荐文章于 2024-03-13 22:50:53 发布

纷繁中淡定

最新推荐文章于 2024-03-13 22:50:53 发布

阅读量318

点赞数

分类专栏：深度学习（pytorch）

本文链接：https://blog.csdn.net/qq_36170626/article/details/104261753

版权

本文介绍了MXNET中的数据操作，重点讲解了NDArray类的功能，包括创建、运算、索引及与NumPy的转换。此外，还探讨了自动求取梯度的机制，以及如何在MXNet中使用autograd模块进行训练模式和预测模式的切换。最后提到了查阅文档的方法，如使用dir函数来查找模块中的函数和类。

摘要由CSDN通过智能技术生成

MXNET中NDArray类：存储和变换数据的主要工具，提供GPU计算和自动求梯度功能，与NumPy的多维数组类似。

导入模块也称作包含的头文件(nd是ndarray的缩写)

from mxnet import nd

X=x.reshape((3,4))   #生成3*4的矩阵

nd.zeros((2,3,4)) #生成两组3*4的矩阵,各个元素都为0
nd.ones(3,4)      #生成元素为1的矩阵

随机生成NDArray中每个元素的值

nd.random.normal(0,1,shape=(3,4))    #均值为0，方差为1的正态分布3*4矩阵

支持加减乘除指数等运算（乘法为对应元素相乘）

矩阵乘法（需要保证维度一致）

关注