onnx model output dimension.

[YaoJiawei329]

Hello! I am confused with the onnx model I exported. I wonder whether the 'masks' node's loss the first dimension or not. As far as I am concerned, it should be [1, 100, 160, 160], but not [100, 160, 160], which means it loss the batch dim.