Skip to content

Commit 9c8f2bb

Browse files
committed
Add gradInput nil check
1 parent 0e545aa commit 9c8f2bb

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

SplitTable.lua

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,13 @@ end
3131
function SplitTable:updateGradInput(input, gradOutput)
3232
local dimension = self:_getPositiveDimension(input)
3333
local slices = input:size(dimension)
34-
self.gradInput:resizeAs(input)
34+
if self.gradInput then
35+
self.gradInput:resizeAs(input)
3536

36-
for i=1,slices do
37-
local currentGradInput = gradOutput[i];
38-
self.gradInput:select(dimension,i):copy(currentGradInput)
37+
for i=1,slices do
38+
local currentGradInput = gradOutput[i];
39+
self.gradInput:select(dimension,i):copy(currentGradInput)
40+
end
3941
end
4042
return self.gradInput
4143
end

0 commit comments

Comments
 (0)