ind

Author: xadupre

_unittests/ut_reference/test_torch_evaluator.py

@@ -444,13 +444,13 @@ def test_op_reduce_sum(self):
             model,
             torch.rand(3, 4, 5, dtype=torch.float32),
             torch.tensor([1], dtype=torch.int64),
-            atol=1e-6,
+            atol=1e-5,
         )
         self._finalize_test(
             model,
             torch.rand(3, 4, 5, dtype=torch.float32),
             torch.tensor([1, 2], dtype=torch.int64),
-            atol=1e-6,
+            atol=1e-5,
         )
 
 

onnx_diagnostic/reference/torch_ops/access_ops.py

@@ -18,7 +18,7 @@ def run(self, x, indices):
             return torch.empty((0,), dtype=x.tensor.dtype, device=x.tensor.device)
         ind = [slice(0, s) for s in x.shape]
         ind[self.axis] = indices.tensor
-        return OpRunValue(x.tensor[*ind])
+        return OpRunValue(x.tensor[tuple(ind)])
 
 
 class Slice_13(OpRun):