From bbe044f3853fcd2fb81d9652aaa03723ae6232e4 Mon Sep 17 00:00:00 2001
From: nihui <shuizhuyuanluo@126.com>
Date: Sat, 15 Dec 2018 12:59:26 +0800
Subject: [PATCH] convert onnx Flatten

---
 tools/onnx/onnx2ncnn.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp
index abe8f176dff..fdda157f25d 100644
--- a/tools/onnx/onnx2ncnn.cpp
+++ b/tools/onnx/onnx2ncnn.cpp
@@ -556,6 +556,10 @@ int main(int argc, char** argv)
         {
             fprintf(pp, "%-16s", "UnaryOp");
         }
+        else if (op == "Flatten")
+        {
+            fprintf(pp, "%-16s", "Flatten");
+        }
         else if (op == "Floor")
         {
             fprintf(pp, "%-16s", "UnaryOp");
@@ -1081,6 +1085,14 @@ int main(int argc, char** argv)
             int op_type = 7;
             fprintf(pp, " 0=%d", op_type);
         }
+        else if (op == "Flatten")
+        {
+            int axis = get_node_attr_i(node, "axis", 1);
+            if (axis != 1)
+            {
+                fprintf(stderr, "Unsupported Flatten axis %d!\n", axis);
+            }
+        }
         else if (op == "Floor")
         {
             int op_type = 2;