pytorch

3c5e3966 - [ONNX] Squeeze operator should give an error when trying to apply to a dimension with shape > 1 (#38476)

[ONNX] Squeeze operator should give an error when trying to apply to a dimension with shape > 1 (#38476) Summary: The ONNX spec for the Squeeze operator: > Remove single-dimensional entries from the shape of a tensor. Takes a parameter axes with a list of axes to squeeze. If axes is not provided, all the single dimensions will be removed from the shape. If an axis is selected with shape entry not equal to one, an error is raised. Currently, as explained in issue https://github.com/pytorch/pytorch/issues/36796, it is possible to export such a model to ONNX, and this results in an exception from ONNX runtime. Fixes https://github.com/pytorch/pytorch/issues/36796. Pull Request resolved: https://github.com/pytorch/pytorch/pull/38476 Reviewed By: hl475 Differential Revision: D22158024 Pulled By: houseroad fbshipit-source-id: bed625f3c626eabcbfb2ea83ec2f992963defa19