Merge pull request #580 from dkurt:onnx_upsample_unsqueeze

Author: alalek

testdata/dnn/onnx/data/input_resize_nearest.npy

@@ -246,3 +246,20 @@ def forward(self, image):
 input = Variable(torch.randn(1, 2, 3, 4))
 model = DynamicReshapeNet()
 save_data_and_model("dynamic_reshape", input, model)
+
+input = Variable(torch.randn(1, 2, 3, 4))
+resize = nn.Upsample(scale_factor=2, mode='nearest')
+save_data_and_model("resize_nearest", input, resize)
+
+class Unsqueeze(nn.Module):
+
+    def __init__(self):
+        super(Unsqueeze, self).__init__()
+
+    def forward(self, x):
+        return torch.unsqueeze(x, dim=1)
+
+input = Variable(torch.randn(1, 2, 3))
+model = Unsqueeze()
+model.eval()
+save_data_and_model("unsqueeze", input, model)

testdata/dnn/onnx/data/input_unsqueeze.npy

@@ -0,0 +1,15 @@
+pytorch0.4:h
+
+
0
1"	Unsqueeze*
+axes@
�
torch-jit-exportZ
+
0
+

+

+
+b
+
1
+

+

+

+
+B