From 6ee12a28e5d7417d1d5908e4cbd20707bddb6711 Mon Sep 17 00:00:00 2001
From: Clement Farabet <clement.farabet@gmail.com>
Date: Wed, 13 Jul 2011 12:35:35 -0400
Subject: Added convenient empty function()

---
 init.lua | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

(limited to 'init.lua')

diff --git a/init.lua b/init.lua
index ae4f350..27dfbfd 100644
--- a/init.lua
+++ b/init.lua
@@ -98,3 +98,30 @@ torch.include('nnx', 'StochasticTrainer.lua')
 torch.include('nnx', 'DataSet.lua')
 torch.include('nnx', 'DataList.lua')
 torch.include('nnx', 'DataSetLabelMe.lua')
+
+-- helpers:
+function nnx.empty(module)
+   if module.modules then
+      -- find submodules in classic containers 'modules'
+      for _,module in ipairs(module.modules) do
+         nnx.empty(module)
+      end
+   else
+      -- find arbitrary submodules
+      for k,entry in pairs(module) do
+         local type = torch.typename(entry)
+         if type and type:find('^nn.') then
+            nnx.empty(entry)
+         end
+      end
+   end
+   -- empty module
+   if module.output and module.output.resize then 
+      module.output:resize()
+      module.output:storage():resize(0)
+   end
+   if module.gradInput and module.gradInput.resize then 
+      module.gradInput:resize()
+      module.gradInput:storage():resize(0)
+   end
+end
-- 
cgit v1.2.3