[layers] get_layers_with_name

Author: zsdonghao

docs/modules/layers.rst

@@ -258,6 +258,7 @@ Layer list
 .. autosummary::
 
    get_variables_with_name
+   get_layers_with_name
    set_name_reuse
    print_all_variables
    initialize_global_variables
@@ -340,6 +341,10 @@ Get variables with name
 ^^^^^^^^^^^^^^^^^^^^^^^^^^
 .. autofunction:: get_variables_with_name
 
+Get layers with name
+^^^^^^^^^^^^^^^^^^^^^^^^^^
+.. autofunction:: get_layers_with_name
+
 Enable layer name reuse
 ^^^^^^^^^^^^^^^^^^^^^^^^^
 .. autofunction:: set_name_reuse

tensorlayer/layers.py

@@ -181,6 +181,28 @@ def get_variables_with_name(name, train_only=True, printable=False):
             print("  got {:3}: {:15}   {}".format(idx, v.name, str(v.get_shape())))
     return d_vars
 
+def get_layers_with_name(network=None, name="", printable=False):
+    """Get layer list by a given name scope.
+
+    Examples
+    ---------
+    >>> layers = tl.layers.get_layers_with_name(network, "CNN", True)
+    """
+    assert network is not None
+    print("  [*] geting layers with %s" % name)
+
+    layers = []
+    i = 0
+    for layer in network.all_layers:
+        # print(type(layer.name))
+        if name in layer.name:
+            layers.append(layer)
+            if printable:
+                # print(layer.name)
+                print("  got {:3}: {:15}   {}".format(i, layer.name, str(layer.get_shape())))
+                i = i + 1
+    return layers
+
 def list_remove_repeat(l=None):
     """Remove the repeated items in a list, and return the processed list.
     You may need it to create merged layer like Concat, Elementwise and etc.