diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-12 02:36:15 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-02-12 02:36:15 +0300 |
commit | 06ff3228dfc4ae32eb2bb55b30ac39aec095e0db (patch) | |
tree | 44c3323019f3eb82ec4beb665ab939bb03317d90 /MulConstant.lua | |
parent | d0a0e9ba6b68fd8392f37e1ac0ad2289260baedf (diff) |
fix direct assignment in some modules
Diffstat (limited to 'MulConstant.lua')
-rw-r--r-- | MulConstant.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/MulConstant.lua b/MulConstant.lua index 0c7f993..e8c473b 100644 --- a/MulConstant.lua +++ b/MulConstant.lua @@ -15,7 +15,7 @@ end function MulConstant:updateOutput(input) if self.inplace then input:mul(self.constant_scalar) - self.output = input + self.output:set(input) else self.output:resizeAs(input) self.output:copy(input) @@ -28,7 +28,7 @@ function MulConstant:updateGradInput(input, gradOutput) if self.gradInput then if self.inplace then gradOutput:mul(self.constant_scalar) - self.gradInput = gradOutput + self.gradInput:set(gradOutput) -- restore previous input value input:div(self.constant_scalar) else |