在深度学习中,我们通常会频繁地对数据进行操作。可以说,数据处理是深度学习的基础。为了开始我们后续的工作,首先我们需要存储和处理数。如果您之前使用过Python的科学计算包NumPy,那么您将会接下来的内容非常熟悉。
在DJL的NDArray不仅支持CUP,还支持GPU以及分离式云端结构,并且支持自动微分等功能。
新手上路
新手上路,我们先用arange创建一个一个行向量x,包含从0开始连续的12个整数。
NDManager manager = NDManager.newBaseManager();
NDArray x = manager.arange(12);
System.out.println(x);
//输出值为:
ND: (12)