✨[Feature] add support for aten::Index & aten::index_select & aten::index_put_

[kiddyjinjin]

aten::index && aten::index_put_

• Function Schema:
  aten::index.Tensor(Tensor self, Tensor?[] indices) -> Tensor
aten::index_put(Tensor self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor
aten::index_put_(Tensor(a!) self, Tensor?[] indices, Tensor values, bool accumulate=False) -> Tensor(a!)
• Tensor Index API:
  https://pytorch.org/cppdocs/notes/tensor_indexing.html

aten::index_select

• Function Schema:
  aten::index_select(Tensor self, int dim, Tensor index) -> Tensor
• Original PyTorch API:
  https://pytorch.org/docs/stable/generated/torch.index_select.html

[ErenBalatkan]

+1, very important feature for some of the transformer models.

[github-actions]

This issue has not seen activity for 90 days, Remove stale label or comment or this will be closed in 10 days