【深度学习笔记】2.2数据操作

本文详细介绍了深度学习框架MXNet中的NDArray,包括创建、运算、广播机制、索引、内存管理和与NumPy的交互。NDArray作为数据存储和处理的核心,支持GPU计算和自动求梯度,其运算符类似于NumPy,但提供了更多的深度学习功能。
摘要由CSDN通过智能技术生成

【深度学习笔记】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]<
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值