文章目录
本博客参考《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))