diff --git a/mindspore/nn/layer/normalization.py b/mindspore/nn/layer/normalization.py
index 6e9236955088a57b03428707b2be771835212ed5..2a1ca28ed45b184001137f0afcd72eca729eb032 100644
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@@ -119,7 +119,7 @@ class _BatchNorm(Cell):
     def _shape_infer(self, x):
         """global batch normalization shape and axes infer"""
         if len(self.shape(x)) == 4:
-            axes = (0,2,3)
+            axes = (0, 2, 3)
             re_shape = (1, self.num_features, 1, 1)
         else:
             axes = (0,)