fix infinite recursion in PullTable and PushTable when converting to different types

1 files changed, 17 insertions, 0 deletions

diff --git a/PullTable.lua b/PullTable.lua
index 1fed379..1413a07 100644
--- a/PullTable.lua
+++ b/PullTable.lua
@@ -58,3 +58,20 @@ function PullTable:updateGradInput(inputTable, gradOutputTable)
    end
    return self.gradInput
 end
+
+
+function PullTable:type(type, tensorCache)
+   assert(type, 'PullTable: must provide a type to convert to')
+
+   tensorCache = tensorCache or {}
+
+   -- find all tensors and convert them
+   for key,param in pairs(self) do
+       if(key ~= "_push") then
+             self[key] = nn.utils.recursiveType(param, type, tensorCache)
+   	     end
+   end
+
+   return self
+end
+