diff --git a/source/operator/prototype/flatten.c b/source/operator/prototype/flatten.c
index 5354dfce26d101b13ba0342ce9b7f5d8302e416e..11536cc1f9bc28efff9577584ca5acedff867cd7 100644
--- a/source/operator/prototype/flatten.c
+++ b/source/operator/prototype/flatten.c
@@ -53,7 +53,9 @@ static int infer_shape(struct node* node)
 
     output->layout = TENGINE_LAYOUT_NHWC;
 
-    set_ir_tensor_shape(output, dims, 4);
+    //set_ir_tensor_shape(output, dims, 4);
+    //stephen@20220923
+    set_ir_tensor_shape(output, dims, 2);
 
     return 0;
 }