fixing bug in ConcatTable variable length

1 files changed, 4 insertions, 1 deletions

diff --git a/ConcatTable.lua b/ConcatTable.lua
index 93d9ad5..7d49af1 100644
--- a/ConcatTable.lua
+++ b/ConcatTable.lua
@@ -14,13 +14,16 @@ function ConcatTable:updateOutput(input)
 end
 
 local function retable(t1, t2, f)
-   for k, v in pairs(t2) do
+   for k, v in ipairs(t2) do
       if (torch.type(v) == "table") then
          t1[k] = retable(t1[k] or {}, t2[k], f)
       else
          f(t1, k, v)
       end
    end
+   for i=#t2+1, #t1 do
+      t1[i] = nil
+   end
    return t1
 end