Fix Python IndexError of Case14: paddle.nn.functional.glu

python/paddle/tensor/manipulation.py

@@ -1918,6 +1918,7 @@ def split(x, num_or_sections, axis=0, name=None):
             dim = dim.numpy()
             dim = dim.item(0)
         assert len(input.shape) + dim >= 0, "(rank(x) + axis) must >= 0"
+        assert dim < len(input.shape), "axis must < (rank(x)"
         dim = (len(input.shape) + dim) if dim < 0 else dim
 
         if isinstance(num_or_sections, (list, tuple)):
@@ -1997,6 +1998,7 @@ def _get_SectionsTensorList(one_list):
             inputs['AxisTensor'] = dim
         else:
             assert len(input.shape) + dim >= 0, "(rank(x) + axis) must >= 0"
+            assert dim < len(input.shape), "axis must < (rank(x)"
             dim = (len(input_shape) + dim) if dim < 0 else dim
             attrs['axis'] = dim