fix shape

justinchuby · justinchuby · commit 3b406aac8b7c · 2025-08-27T10:33:34.000-07:00
Signed-off-by: Justin Chu &lt;justinchuby@users.noreply.github.com&gt;
diff --git a/onnxscript/function_libs/torch_lib/ops/common.py b/onnxscript/function_libs/torch_lib/ops/common.py
@@ -102,7 +102,7 @@ def merge_dims(dims: Sequence[int | INT64]) -> INT64:
         else:
             # A dynamic dimension, unsqueeze and append it
             current_dim = op.Reshape(
-                current_dim, op.Constant(value_ints=ir.AttrInt64("value_ints", []))
+                current_dim, op.Constant(value_ints=ir.AttrInt64("value_ints", [1]))
             )
             result_dims.append(current_dim)
     if len(result_dims) == 1:

Original file line number	Diff line number	Diff line change
`@@ -102,7 +102,7 @@ def merge_dims(dims: Sequence[int \| INT64]) -> INT64:`
`102`	`102`	`else:`
`103`	`103`	`# A dynamic dimension, unsqueeze and append it`
`104`	`104`	`current_dim = op.Reshape(`
`105`		`- current_dim, op.Constant(value_ints=ir.AttrInt64("value_ints", []))`
	`105`	`+ current_dim, op.Constant(value_ints=ir.AttrInt64("value_ints", [1]))`
`106`	`106`	`)`
`107`	`107`	`result_dims.append(current_dim)`
`108`	`108`	`if len(result_dims) == 1:`