MxNet 学习笔记(1):MxNet中的NDArray
http://mxnet.incubator.apache.org/api/python/symbol/symbol.html api文档
MXNet API入门 —第6篇
MXNet API入门 —第5篇
MXNet API入门 —第4篇
MXNet API入门 —第3篇
MXNet API入门 —第2篇
MXNet API入门 —第1篇
[MXNet代码剖析] NNVM计算图抽象机制
查看mxnet中的symbol图的所有变量,以及他们的shape
>>> import mxnet as mx
>>>
>>> a = mx.sym.Variable('data') >>> b = mx.sym.FullyConnected(data=a,name='fc1',num_hidden=100) >>> data_shape = {'data':(256,64)} >>> arg_shape,_,_ = b.infer_shape(**data_shape) >>> b.list_arguments() # 列出symbol中的所有参数,这里是输入以及全连接层的权值和偏置 ['data', 'fc1_weight', 'fc1_bias'] >>> arg_shape # 这里展示上面三个参数的size [(256L, 64L), (100L, 64L), (100L,)]
参考:https://github.com/apache/incubator-mxnet/issues/5271