add "update_by_batch" in gene;

add flatten_conns as an inverse function for unflatten_conns;
add "test_flatten.ipynb" as test for them.

tensorneat/algorithm/neat/genome/base.py

@@ -120,3 +120,9 @@ class BaseGenome:
         conns = conns.at[: len(conn_keys), 3:].set(random_conn_attrs)
 
         return nodes, conns
+
+    def update_by_batch(self, state, batch_input, nodes, conns):
+        """
+        Update the genome by a batch of data.
+        """
+        raise NotImplementedError

tensorneat/algorithm/neat/genome/default.py

@@ -93,3 +93,6 @@ class DefaultGenome(BaseGenome):
             return vals[self.output_idx]
         else:
             return self.output_transform(vals[self.output_idx])
+
+    def update_by_batch(self, state, batch_input, nodes, conns):
+        pass