api文档

 

 

查看mxnet中的symbol图的所有变量，以及他们的shape

2018年03月13日 19:26:21

阅读数：304

>>> 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,)]

参考：