Merge pull request #53 from zhangxiangxiao/master

Added share function for gModule

1 files changed, 9 insertions, 0 deletions

diff --git a/gmodule.lua b/gmodule.lua
index e6e5fab..3021855 100644
--- a/gmodule.lua
+++ b/gmodule.lua
@@ -363,6 +363,15 @@ function gModule:parameters()
 	return p,gp
 end
 
+-- Added share function
+function gModule:share(gModuleToShare, ...)
+   for indexNode, node in ipairs(self.forwardnodes) do
+      if node.data.module then
+	 node.data.module:share(gModuleToShare.forwardnodes[indexNode].data.module, ...)
+      end
+   end
+end
+
 function gModule:__tostring__()
 	return self.name or torch.type(self)
 end