[ONNX] Support nd mask index in opset >= 11 (#45252)
Summary:
Fixes below pattern for opset >= 11
`return tensor[tensor > 0]`
where rank of `tensor` > 1.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45252
Reviewed By: VitalyFedyunin
Differential Revision: D24116945
Pulled By: bzinodev
fbshipit-source-id: 384026cded1eb831bb5469e31ece4fcfb6ae8f2a