首页 > Web开发 > 详细

mxnet 查看 Sym shape

时间:2019-01-07 20:28:13      阅读:281      评论:0      收藏:0      [点我收藏+]

import mxnet as mx
import numpy as np
import random
import mxnet as mx
import sys
data_shape = {‘data‘:(60000, 1,28, 28)}
data = mx.sym.var(‘data‘)
pool0 = mx.sym.Pooling(data=data, pool_type="max", kernel=(2,2), stride=(2,2),name=‘pool0‘)
pool1 = mx.sym.Pooling(data=pool0, pool_type="max", kernel=(2,2), stride=(2,2),name=‘pool1‘)
in_shape,out_shape,uax_shape = pool1.infer_shape(**data_shape)
#pool0.list_outputs()
in_shape,out_shape,uax_shape

mxnet 查看 Sym shape

原文:https://www.cnblogs.com/jukan/p/10234546.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!