Module:listModules()

1 files changed, 24 insertions, 0 deletions

diff --git a/Module.lua b/Module.lua
index c9b73cc..bc5a7fb 100644
--- a/Module.lua
+++ b/Module.lua
@@ -287,3 +287,27 @@ function Module:findModules(typename, container)
   return nodes, containers
 end
 
+-- returns a list of modules
+function Module:listModules()
+   local function tinsert(to, from)
+      if torch.type(from) == 'table' then
+         for i=1,#from do
+            tinsert(to,from[i])
+         end
+      else
+         table.insert(to,from)
+      end
+   end
+   -- include self first
+   local modules = {self}
+   if self.modules then
+      for i=1,#self.modules do
+         local modulas = self.modules[i]:listModules()
+         if modulas then
+            tinsert(modules,modulas)
+         end
+      end
+   end
+   return modules
+end
+