Fix Expand (again) (#12)

* Update optimizer.py * simplify code --------- Co-authored-by: inisis <[email protected]>
inisis · Jun 19, 2024 · 87e491b · 87e491b
1 parent 44e5912
commit 87e491b
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/onnxslim/core/optimizer.py b/onnxslim/core/optimizer.py
@@ -157,10 +157,8 @@ def graph_constant_fold_inplace(graph):
                     delete_node(node, idx)
                     logger.debug(f"removing Add op: {node.name}")
         elif node.op == "Expand":
-            idx, constant_variable = get_constant_variable(node, return_idx=True)
             if len(node.inputs) > 1 and isinstance(node.inputs[1], Constant) and np.all(node.inputs[1].values == 1):
-                idx = 0 if idx == 1 else 1
-                delete_node(node, idx)
+                delete_node(node)
                 logger.debug(f"removing Expand op: {node.name}")
         elif node.op == "Concat":
             if len(node.inputs) == 1: