[Split] bug fix

jiazhihao · Oct 25, 2019 · 68b24d3 · 68b24d3
1 parent dbbbae2
commit 68b24d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/core/split.cc b/src/core/split.cc
@@ -133,7 +133,7 @@ Split::Split(Model* _model, const Tensor& _input,
       } else {
         outputs[i].dim[j] = _sizes[i];
         outputs[i].stride[j] = inputs[0].stride[j];
-        if (i > 0) {
+        if (i > 0 && parent.num > 0) {
           parent.divide(left, right, curPos);
         } else {
           curPos = 0;