Add gradInput nil check

stephenrawls · May 12, 2016 · 9c8f2bb · 9c8f2bb
1 parent 0e545aa
commit 9c8f2bb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/SplitTable.lua b/SplitTable.lua
@@ -31,11 +31,13 @@ end
 function SplitTable:updateGradInput(input, gradOutput)
    local dimension = self:_getPositiveDimension(input)
    local slices = input:size(dimension)
-   self.gradInput:resizeAs(input)
+   if self.gradInput then
+      self.gradInput:resizeAs(input)
 
-   for i=1,slices do 
-      local currentGradInput = gradOutput[i];        
-      self.gradInput:select(dimension,i):copy(currentGradInput)
+      for i=1,slices do 
+         local currentGradInput = gradOutput[i];        
+         self.gradInput:select(dimension,i):copy(currentGradInput)
+      end
    end
    return self.gradInput
 end