Some suggestion

[huangh12]

It's an awesome work！
It will be better if the label shape could also be infered as the image shape. At now, it is limited to [1], which only caters to the classification application. However, semantic segmentation is a hot topic today where the input label map has the same size as input image. So, the support to the shape infering of label map (not only the image/feature map) will make netscope even much better! :)