Merge pull request #808 from l-bat:issue_18350

Author: alalek

testdata/dnn/onnx/data/input_scale.npy

@@ -977,3 +977,12 @@ def forward(self, x, skip=None):
 x = Variable(torch.rand(1, 2, 2, 2))
 model = ResizeConv(2, 0, 2)
 save_data_and_model("resize_opset11_torch1.6", x, model, 11)
+
+class Scale(nn.Module):
+  def forward(self, x):
+    w = torch.mean(x, axis=(2, 3), keepdim=True)
+    return w * x
+
+x = Variable(torch.randn(1, 3, 2, 2))
+model = Scale()
+save_data_and_model("scale", x, model)

testdata/dnn/onnx/data/output_scale.npy

@@ -0,0 +1,21 @@
+pytorch1.6:�

+@
+
0
1ReduceMean_0"
+ReduceMean*
+axes@@�
*
+keepdims
�

+
+
1
+
0
2Mul_1"Multorch-jit-exportZ
+
0
+

+

+
+
+b
+
2
+

+

+
+
+B