https://github.com/microsoft/onnxscript/blob/854b5d958ed8de629d5e273c…

…e1f3ad11358331d1/onnxscript/function_libs/torch_lib/ops/core.py#L8086-L8088
microsoft · Jan 2, 2025 · dbe67e8 · dbe67e8
1 parent 854b5d9
commit dbe67e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -8085,7 +8085,7 @@ def aten_swapdims(self: TensorType, dim0: int, dim1: int) -> TensorType:
 @torch_op("aten::sym_size.int", trace_only=True)
 def aten_sym_size(self: TensorType, dim: int = 0) -> INT64:
     """sym_size.int(Tensor self, int dim) -> SymInt"""
-    return op.Shape(self, end=dim + 1, start=dim)
+    return op.Squeeze(op.Shape(self, end=dim + 1, start=dim))
 
 
 def aten_symeig(