From 18175b9eb27afa8b0de76189224ebbca3fe2145b Mon Sep 17 00:00:00 2001
From: zoom <zhongwl2018@mail.sustech.edu.cn>
Date: Fri, 14 Oct 2022 17:02:43 +0800
Subject: [PATCH] add test for issue#22625

---
 .../onnx/data/input_unsqueeze_neg_axes.npy    | Bin 0 -> 152 bytes
 .../onnx/data/output_unsqueeze_neg_axes.npy   | Bin 0 -> 152 bytes
 testdata/dnn/onnx/generate_onnx_models.py     |  11 ++++++++---
 .../dnn/onnx/models/unsqueeze_neg_axes.onnx   |  18 ++++++++++++++++++
 4 files changed, 26 insertions(+), 3 deletions(-)
 create mode 100644 testdata/dnn/onnx/data/input_unsqueeze_neg_axes.npy
 create mode 100644 testdata/dnn/onnx/data/output_unsqueeze_neg_axes.npy
 create mode 100644 testdata/dnn/onnx/models/unsqueeze_neg_axes.onnx

diff --git a/testdata/dnn/onnx/data/input_unsqueeze_neg_axes.npy b/testdata/dnn/onnx/data/input_unsqueeze_neg_axes.npy
new file mode 100644
index 0000000000000000000000000000000000000000..c557d93c23365fb022a31a8275ff4bb89bfc2e5f
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3dWi`3bhL40j?PPqxJzd)Ao55av#{%B51$xV%L6M*3J6?a+D@^

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/data/output_unsqueeze_neg_axes.npy b/testdata/dnn/onnx/data/output_unsqueeze_neg_axes.npy
new file mode 100644
index 0000000000000000000000000000000000000000..89082ef9e08f874898ab41889aff6463954e13b7
GIT binary patch
literal 152
zcmbR27wQ`j$;eQ~P_3SlTAW;@Zl$1ZlV+l>qoAIaUsO_*m=~X4l#&V(cT3DEP6dh=
zXCxM+0{I$-ItoTQ3P93WQ%9jzfegSEV}H~>z-HP$&qD44+gb$e7hdezugki5KLC1~
BCX)aF

literal 0
HcmV?d00001

diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py
index dea373451..a0627e9ba 100644
--- a/testdata/dnn/onnx/generate_onnx_models.py
+++ b/testdata/dnn/onnx/generate_onnx_models.py
@@ -392,18 +392,23 @@ def forward(self, image):
 
 class Unsqueeze(nn.Module):
 
-    def __init__(self):
+    def __init__(self, dim):
         super(Unsqueeze, self).__init__()
+        self.dim = dim
 
     def forward(self, x):
-        return torch.unsqueeze(x, dim=1)
+        return torch.unsqueeze(x, dim = self.dim)
 
 input = Variable(torch.randn(1, 2, 3))
-model = Unsqueeze()
+model = Unsqueeze(1)
 model.eval()
 save_data_and_model("unsqueeze", input, model)
 save_data_and_model("unsqueeze_opset_13", input, model, version=13)
 
+model = Unsqueeze(-2)
+model.eval()
+save_data_and_model("unsqueeze_neg_axes", input, model)
+
 input = Variable(torch.randn(1, 2, 4, 5))
 deconv_adjpad2d = nn.ConvTranspose2d(2, 3, (3, 2), stride=(1, 2), padding=(1, 2), output_padding=(0, 1))
 save_data_and_model("deconv_adjpad_2d", input, deconv_adjpad2d)
diff --git a/testdata/dnn/onnx/models/unsqueeze_neg_axes.onnx b/testdata/dnn/onnx/models/unsqueeze_neg_axes.onnx
new file mode 100644
index 000000000..425daf0d9
--- /dev/null
+++ b/testdata/dnn/onnx/models/unsqueeze_neg_axes.onnx
@@ -0,0 +1,18 @@
+pytorch1.12.1:Û

+Eonnx::Unsqueeze_1
+Constant_0"Constant*
+value*
Jþÿÿÿÿÿÿÿ 

+A
+onnx::Unsqueeze_0
+onnx::Unsqueeze_1
2Unsqueeze_1"	Unsqueeze	torch_jitZ'
+onnx::Unsqueeze_0
+

+

+
+b
+
2
+

+

+
+

+B
\ No newline at end of file