arg_name = fc1.list_arguments()
out_name = fc1.list_outputs()
arg_shape, out_shape, _ = fc1.infer_shape(data=(1,3,112,112))
print({'input' : dict(zip(arg_name, arg_shape)),'output' : dict(zip(out_name, out_shape))})
mxnet如何打印symbol输出维度
最新推荐文章于 2022-11-05 01:25:33 发布