l.reset_states()
def set_previous(self, layer, connection_map={}):
if self.nb_input != layer.nb_output:
raise Exception("Cannot connect layers: "
"input count does not match output count.")
if self.nb_input == 1:
self.inputs[self.input_order[0]].set_previous(layer)
else:
if not connection_map:
raise Exception("Cannot attach multi-input layer: "
"no connection_map provided.")
for k, v in connection_map.items():
if k in self.inputs and v in layer.outputs:
self.inputs[k].set_previous(layer.outputs[v])
else:
raise Exception("Invalid connection map.")
def get_input(self, train=False):
if len(self.inputs) == len(self.outputs) == 1:
return self.inputs[self.input_order[0]].get_input(train)
After Change
Note: `Graph` can only be used as a layer
(connect, input, get_input, get_output)
when it has exactly one input and one output.
"""
def __init__(self):
self.namespace = set() // strings
self.nodes = OrderedDict() // layer-like
