base.py 文件源码

def _construct_layer_maps(self):
        layers = L.get_all_layers(self.output_layer)
        # Store inverse layers to enable merging.
        self.inverse_map = {l: None for l in layers}
        # Store the layers a specific layer feeds.
        self.output_map = {l: [] for l in layers}

        for layer in  layers:
            if type(layer) is not L.InputLayer:
                if isinstance(layer, L.MergeLayer):
                    for feeder in layer.input_layers:
                        self.output_map[feeder].append(layer)
                else:
                    self.output_map[layer.input_layer].append(layer)