mxnet学习二、基础(了解NDArray)

本博客参考《MXNet深度学习实战》所做的系列笔记

一、学习系列传送门

之前的笔记：
1、mxnet学习一、数据制作以及读取

二、了解NDArray

NDArray是mxnet框架中数据流的基础结构，和python下的numpy非常相似，可以看做GPU下的numpy。
1、定义NDArray

a = mx.nd.array([[1,2],[4,3]])

输出

[[1. 2.]
 [4. 3.]]
<NDArray 2x2 @cpu(0)>

2、NDArray和numpy互相转换

NDArray转numpy，调用NDArray的asnumpy()对象

a = mx.nd.array([[1,2],[3,4]])
b = a.asnumpy()

numpy转NDArray，利用mx.nd.array()

a = np.array([[1,2],[3,4]])
b = mx.nd.array(a)

3、查看在哪张卡上运行的NDArray

print(b.context)

4、利用NDArray对象的as_in_context()

a = a.as_in_context(mx.gpu(0))