- NDArray:数组,类似于pytorch中的Tensor(封装的numpy的array,使之可以在GPU运行)
- Symbol:模型的层,类似于pytorch中的layers
- Module:模型整体框架
MXNet中的矩阵用法基本和Numpy的array一致:
- 创建矩阵:
a = mx.nd.array([[1,2],[3,4]])
- 查看维度:
a.shape
- 查看数值类型:
a.dtype
- 切片
- 指定数据类型,
,dtype=np.int8
- 数组复制:
copy()
,否则,修改会改变原有的值 - ndarray转换成numpy:
a.asnumpy()
- array转为ndarray:
mx.nd.array(d)