From dbe67e8c1002b94739f9f134ce416799ec98b9cd Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Thu, 2 Jan 2025 15:57:09 -0800 Subject: [PATCH] https://github.com/microsoft/onnxscript/blob/854b5d958ed8de629d5e273ce1f3ad11358331d1/onnxscript/function_libs/torch_lib/ops/core.py#L8086-L8088 --- 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 9de7b170f..6fb230e90 100644 --- 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(