fix

x2paddle/op_mapper/onnx2paddle/opset9/opset.py

@@ -688,16 +688,14 @@ class OpSet9():
         axes = node.get_attr('axes')
         if axes is None:
             axes = self.graph.get_input_node(node, idx=1, copy=True)
-        if node.name in ["x2paddle_vis_local_cost_volume_3d_0_ExpandDims_5_0"]:
-            print("output_shape:", val_x.out_shapes[0])
-        # if len(val_x.out_shapes[0]) == 0:
-        #     if node.name:
-        #         self.paddle_graph.add_layer(
-        #             'paddle.reshape',
-        #             inputs={"x": val_x.name},
-        #             outputs=[node.name],
-        #             shape=[1])
-        # else:
+        if len(val_x.out_shapes[0]) == 0 and len(axes) == 1 and axes[0] == 0:
+            if node.name:
+                self.paddle_graph.add_layer(
+                    'paddle.reshape',
+                    inputs={"x": val_x.name},
+                    outputs=[node.name],
+                    shape=[1])
+        else:
             if isinstance(axes, list) or isinstance(axes, tuple):
                 self.paddle_graph.add_layer(
                     'paddle.unsqueeze',