import tensorflow as tf

            tf.add_to_collection('mu_sigma_bn', sigma)
            beta = tf.get_variable('beta', batch_mean.shape, dtype=tf.float32,
                                   initializer=tf.zeros_initializer())
            gamma = tf.get_variable('gamma', batch_var.shape, dtype=tf.float32,
