Fix Python IndexError of case11: paddle.static.nn.data_norm (#50097)

* 为data_norm函数的input参数增加维度的判断并增加单侧

* 完善data_norm测试样例

* fix data_norm

python/paddle/fluid/tests/unittests/test_data_norm_op.py

@@ -523,6 +523,11 @@ class TestDataNormOpErrorr(unittest.TestCase):
                 input=x2, param_attr={}, enable_scale_and_shift=True
             )
 
+            # Test input with dimension 1
+            paddle.enable_static()
+            x3 = paddle.static.data("", shape=[0], dtype="float32")
+            self.assertRaises(ValueError, paddle.static.nn.data_norm, x3)
+
 
 if __name__ == '__main__':
     unittest.main()

python/paddle/static/nn/common.py

@@ -514,6 +514,12 @@ def data_norm(
     dtype = helper.input_dtype()
 
     input_shape = input.shape
+    if len(input_shape) < 2:
+        raise ValueError(
+            "The shape pf Input < 2 (got {}D input, input shape is: {})".format(
+                len(input_shape), input_shape
+            )
+        )
     if data_layout == 'NCHW':
         channel_num = input_shape[1]
     else: