From 361af04dca7058b28cae5dbc4615fb2753a9a5f6 Mon Sep 17 00:00:00 2001 From: titaiwangms Date: Mon, 10 Mar 2025 20:54:32 +0000 Subject: [PATCH] improve dim expression --- onnxscript/function_libs/torch_lib/ops/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py index c0728c833c..a78e4a7ff9 100644 --- a/onnxscript/function_libs/torch_lib/ops/core.py +++ b/onnxscript/function_libs/torch_lib/ops/core.py @@ -7709,7 +7709,7 @@ def aten_slice_complex( """slice.Tensor(Tensor(a) self, int dim=0, SymInt? start=None, SymInt? end=None, SymInt step=1) -> Tensor(a)""" if dim < 0: # Account for the complex dimension in ONNX - dim = dim - 1 + dim = len(self.shape) + dim - 1 return aten_slice(self, dim, start, end, step)