added split = 1 test for split layer

Author: Abduragim Shtanchaev

testdata/dnn/onnx/data/input_split_0.npy

@@ -868,6 +868,11 @@ def forward(self, x):
         tup = torch.split(x, self.split_size_sections, self.dim)
         return torch.cat(tup)
 
+
+model = Split(dim=0, split_size_sections=1)
+input = Variable(torch.ones((1, 3, 2, 2), dtype=torch.float32))
+save_data_and_model("split_0", input, model)
+
 model = Split()
 input = Variable(torch.tensor([1., 2.], dtype=torch.float32))
 save_data_and_model("split_1", input, model)
@@ -888,6 +893,8 @@ def forward(self, x):
 model = Split(dim=-1, split_size_sections=[1, 2])
 save_data_and_model("split_6", input2, model, version=13)
 
+
+
 class SplitSizes(nn.Module):
     def __init__(self, *args, **kwargs):
         super(SplitSizes, self).__init__()