如果要打印输出层维度也很简单,代码如下:
mx.viz.print_summary(sym, shape={"data":(1, 1, 32, 32)})
——————————————————————————————————————
对于HybridSequential类的输出维度打印,方法如下:
mx.viz.print_summary(
net(mx.sym.var('data')),
shape={'data':(1,3,224,224)}, #set your shape here
)
————————————————
版权声明:本文为CSDN博主「zhqh100」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/zhqh100/article/details/89876604