超级会员免费看
本文介绍了在Python中如何使用MXNet的np和npx模块进行张量操作，特别是张量的创建和初始化。通过arange创建行向量，并讨论了张量的元素概念。此外，还讲解了如何初始化张量为1或0，以及如何生成包含随机数的张量，这些操作在深度学习开发中至关重要。
MXNet导入np（numpy）模块和npx（numpy_extension）模块。 np模块包含NumPy支持的函数； 而npx模块包含一组扩展函数，用来在类似NumPy的环境中实现深度学习开发。 当使用张量时，几乎总是会调用set_np函数，这是为了兼容MXNet的其他张量处理组件。
首先，我们可以使用 arange 创建一个行向量 x。这个行向量包含以0开始的前12个整数，它们默认创建为浮点数。张量中的每个值都称为张量的 元素（element）。例如，张量 x 中有 12 个元素。除非额外指定，新的张量将存储在内存中，并采用基于CPU的计算。
通过一个例子来看看：
from mxnet import np, npx
npx.set_np()
x = np
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
