From f47fbcbf83141f37827a86cd5c6f927f93ebd007 Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Wed, 3 Feb 2021 15:27:45 +0800
Subject: [PATCH] preserve the very first onnx constant dimension size when we
 guess it is not the batchsize one, fix #2487

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

diff --git a/tools/onnx/onnx2ncnn.cpp b/tools/onnx/onnx2ncnn.cpp
index ff64075e7..2c3313b70 100644
--- a/tools/onnx/onnx2ncnn.cpp
+++ b/tools/onnx/onnx2ncnn.cpp
@@ -2606,11 +2606,19 @@ int main(int argc, char** argv)
         else if (M.dims_size() == 2)
         {
             fprintf(pp, " 0=%d", (int)M.dims(1));
+            if (M.dims(0) != 1)
+            {
+                fprintf(pp, " 1=%d", (int)M.dims(0));
+            }
         }
         else if (M.dims_size() == 3)
         {
             fprintf(pp, " 0=%d", (int)M.dims(2));
             fprintf(pp, " 1=%d", (int)M.dims(1));
+            if (M.dims(0) != 1)
+            {
+                fprintf(pp, " 2=%d", (int)M.dims(0));
+            }
         }
         else if (M.dims_size() == 4)
         {