From 0ba35eaec3c73867056e05259e36b44520355535 Mon Sep 17 00:00:00 2001
From: zhaojichen <zhaojichen1@huawei.com>
Date: Tue, 28 Apr 2020 21:09:15 -0400
Subject: [PATCH] fix globalbatchnorm bug

---
 mindspore/nn/layer/normalization.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mindspore/nn/layer/normalization.py b/mindspore/nn/layer/normalization.py
index 9d623bc6fd..66f17e3f38 100644
--- a/mindspore/nn/layer/normalization.py
+++ b/mindspore/nn/layer/normalization.py
@@ -117,6 +117,7 @@ class _BatchNorm(Cell):
         return group_list
 
     def _global_sync(self, x):
+        """calculate global batch normalization output"""
         if len(self.shape(x)) == 4:
             axes = (0, 2, 3)
             re_shape = (1, self.num_features, 1, 1)