Fix backward-compatibility of nn.View

1 files changed, 2 insertions, 0 deletions

diff --git a/View.lua b/View.lua
index 8a1c6ac..542e57e 100644
--- a/View.lua
+++ b/View.lua
@@ -75,6 +75,7 @@ local function batchsize(input, size, numInputDims, numElements)
 end
 
 function View:updateOutput(input)
+   self.output = self.output or input.new()
    local bsz = batchsize(input, self.size, self.numInputDims, self.numElements)
    if bsz then
       self.output:view(input, bsz, table.unpack(self.size:totable()))
@@ -85,6 +86,7 @@ function View:updateOutput(input)
 end
 
 function View:updateGradInput(input, gradOutput)
+   self.gradInput = self.gradInput or gradOutput.new()
    self.gradInput:view(gradOutput, input:size())
    return self.gradInput
 end