>>> 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,)]
查看mxnet中的symbol图的所有变量,以及他们的shape
最新推荐文章于 2024-07-30 21:05:50 发布