extras.py 文件源码

def find_bn_updates(layers):
    """
        Return a list of tuples of all the bn_layers in this list of layers
    """
    bn_updates = []
    layers = ll.get_all_layers(layers)
    for layer in layers:
        if hasattr(layer, 'bn_updates'):
            bn_updates.extend(layer.bn_updates)
    return bn_updates