网络hybridize之后怎么获取shape


#1

在前向的时候需要获取batch_size和网络的输出维度对结果进行reshape,大概就是(b,c,h,w)->(b,c*h,w)这个操作,这个应该怎么实现?


#2

reshape有一些特定含义的值, 比如-1是根据其他维度自动计算, 中间c和h合并可以参考文档

http://mxnet.incubator.apache.org/api/python/docs/api/ndarray/op/index.html#mxnet.ndarray.op.reshape


#3

多谢,已经用你的方法解决