ONNX support for torch.take (#33061)
Summary:
Adding ONNX export support for torch.take()
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33061
Reviewed By: hl475
Differential Revision: D19782651
Pulled By: houseroad
fbshipit-source-id: 0168fb941e166acda4ca607165248b8e0b260ace