Revert "State variable pointers should never change."

1 files changed, 2 insertions, 2 deletions

diff --git a/Identity.lua b/Identity.lua
index 18a3c9b..088cc34 100644
--- a/Identity.lua
+++ b/Identity.lua
@@ -1,12 +1,12 @@
 local Identity, _ = torch.class('nn.Identity', 'nn.Module')
 
 function Identity:updateOutput(input)
-   self.output:set(input:storage(), input:storageOffset(), input:size(), input:stride())
+   self.output = input
    return self.output
 end
 
 
 function Identity:updateGradInput(input, gradOutput)
-   self.gradInput:set(gradOutput:storage(), gradOutput:storageOffset(), gradOutput:size(), gradOutput:stride())
+   self.gradInput = gradOutput
    return self.gradInput
 end