onnx issue

[wzp8023391]

Thanks for the authors' nice work, I used this code to translate pth file into onnx file, and I got the shape of mask is [1, 1, 1, 3], why did this happend? can you give me some advice?