import tensorflow as tf

            tf.add_to_collection(tf.GraphKeys.GLOBAL_VARIABLES, mu)
            tf.add_to_collection('mu_sigma_bn', mu)
            sigma = tf.get_variable('sigma', batch_var.shape, dtype=tf.float32,
                                    initializer=tf.ones_initializer(), trainable=False)
            tf.add_to_collection(tf.GraphKeys.GLOBAL_VARIABLES, sigma)
            tf.add_to_collection('mu_sigma_bn', sigma)
            beta = tf.get_variable('beta', batch_mean.shape, dtype=tf.float32,
                                   initializer=tf.zeros_initializer())
