[onnx] Fix onnx-to-torch lowering for flatten shape (#2834)

[onnx] Fix onnx-to-torch lowering for flatten shape (#2834) #9