【深度学习笔记】2.2数据操作
在MXNet中,NDArray是⼀个类,也是存储和变换数据的主要⼯具。为了简洁,本书常将NDArray实例直接称作NDArray。如果你之前⽤过NumPy,你会发现NDArray和NumPy的多维数组⾮常类似。然而,NDArray提供GPU计算和⾃动求梯度等更多功能,这些使NDArray更加适合深度学习。
2.2.1创建NDArray
⾸先从MXNet导⼊ndarray模块。这⾥的nd是ndarray的缩写形式。
from mxnet import nd
然后我们⽤arange函数创建⼀个⾏向量。
x = nd.arange(12)
x.shape #通过shape属性来获取NDArray实例的形状
x.size #通过size属性得到NDArray实例中元素(element)的总数
X = x.reshape((3, 4)) #使⽤reshape函数把⾏向量x的形状改为(3, 4),也就是⼀个3⾏4列的矩阵,并记作X。除了形状改变之外,X中的元素保持不变。
nd.zeros((2, 3, 4)) #创建⼀个各元素为0,形状为(2, 3, 4)的张量。实际上,之前创建的向量和矩阵都是特殊的张量。
nd.ones((3, 4)) #创建各元素为1的张量
Y = nd.array([[2, 1, 4, 3]<