Update on "Implement aten::index | feat(torchlib) (#862)"

---

**This change implements the logic for `aten::index` and adds tests for different nd index combinations and permutations.**

## Understanding `aten::index`

For `arg0` with shape `[7, 3, 4, 5, 6]`
The indexing operation `arg0[0, :, 1:2, tensor([[4,5]])]` will be translated to
```
+>  select: i64[3, 4, 5, 6] = torch.ops.aten.select.int(arg0, 0, 0);
+>  slice_1: i64[3, 4, 5, 6] = torch.ops.aten.slice.Tensor(select, 0, 0, 9223372036854775807);
+>  slice_2: i64[3, 1, 5, 6] = torch.ops.aten.slice.Tensor(slice_1, 1, 1, 2);
+>  index: i64[3, 1, 1, 2, 6] = torch.ops.aten.index.Tensor(slice_2, [None, None, arg1]);
```
Here,
- `indices = [None, None, arg1]` is equivalent to `indices = [None, None, arg1, None]`
- The operation `arg0[0, :, 1:2, tensor([[4,5]])]` is equivalent to `arg0[0, :, 1:2, tensor([[4,5]]), :]`
None in `indices` are like fillers for dimensions that cannot be removed in the process.

## Gather op reference

- https://github.com/openxla/xla/blob/main/docs/operation_semantics.md?rgh-link-date=2023-07-13T01%3A09%3A16Z#gather
- https://www.pathpartnertech.com/gather-scatter-operation-in-deep-learning-framework/

---------

Co-authored-by: BowenBao <bowbaomicrosoft.com>

[ghstack-poisoned]

Author: justinchuby

onnxscript/tests/function_libs/torch_lib/extra_opinfo.py

@@ -660,12 +660,11 @@ def sample_inputs_bernoulli_p_deterministic(op_info, device, dtype, requires_gra
         supports_out=False,
     ),
     opinfo_core.OpInfo(
-        "aten.index.Tensor",
+        "ops.aten.index.Tensor",
+        aten_name="index.Tensor",
         dtypes=common_dtype.all_types_and_complex_and(
             torch.bool, torch.float16, torch.bfloat16, torch.chalf
         ),
-        aten_name="index",
-        op=torch.ops.aten.index.Tensor,
         sample_inputs_func=sample_inputs_index,
     ),
     opinfo_core.OpInfo(

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

@@ -607,7 +607,7 @@ def _where_input_wrangler(
     TorchLibOpInfo("gt", core_ops.aten_gt),
     # TorchLibOpInfo("is_same_size", core_ops.aten_is_same_size),  # no test case in OPS_DB
     # TorchLibOpInfo("is_nonzero", core_ops.aten_is_nonzero),  # no test case in OPS_DB
-    TorchLibOpInfo("aten.index.Tensor", core_ops.aten_index, trace_only=True),
+    TorchLibOpInfo("ops.aten.index.Tensor", core_ops.aten_index, trace_only=True),
     TorchLibOpInfo(
         "index_put_bool",
         core_ops.aten_index_put_bool,