ops.py 文件源码

def batch_norm(x, train_mode=True, epsilon=1e-5, momentum=0.9, name="bn"):
    with tf.variable_scope(name):
        return tcl.batch_norm(x, 
                              decay=momentum, 
                              updates_collections=None, 
                              epsilon=epsilon, 
                              scale=True, 
                              is_training=train_mode, 
                              trainable=True, 
                              scope=name)