[Parallel,Multi]Criterion table inputs

1 files changed, 21 insertions, 0 deletions

diff --git a/utils.lua b/utils.lua
index 78a22a2..74ff7e6 100644
--- a/utils.lua
+++ b/utils.lua
@@ -44,6 +44,27 @@ function nn.utils.recursiveFill(t2, val)
    return t2
 end
 
+function nn.utils.recursiveAdd(t1, val, t2)
+   if not t2 then
+      assert(val, "expecting at least two arguments")
+      t2 = val
+      val = 1
+   end
+   val = val or 1
+   if torch.type(t2) == 'table' then
+      t1 = (torch.type(t1) == 'table') and t1 or {t1}
+      for key,_ in pairs(t2) do
+         t1[key], t2[key] = nn.utils.recursiveAdd(t1[key], val, t2[key])
+      end
+   elseif torch.isTensor(t2) and torch.isTensor(t2) then
+      t1:add(val, t2)
+   else
+      error("expecting nested tensors or tables. Got "..
+            torch.type(t1).." and "..torch.type(t2).." instead")
+   end
+   return t1, t2
+end
+
 function nn.utils.addSingletonDimension(t, dim)
   local view = t.new()
   local size = torch.LongStorage(t:dim() + 1)