diff options
author | Jonathan Tompson <tompson@cims.nyu.edu> | 2014-06-25 20:14:12 +0400 |
---|---|---|
committer | Jonathan Tompson <tompson@cims.nyu.edu> | 2014-06-25 20:14:12 +0400 |
commit | e406feabe2b1bd7f6b7a15826aff1e925fe713d6 (patch) | |
tree | 8177ddef413e068c95882e00d228feca0900ebaf /MulConstant.lua | |
parent | ea9cc1df751ddb144c08a13aab3add1ab0ce90a1 (diff) |
Added very simple add and mul constant modules.
Diffstat (limited to 'MulConstant.lua')
-rw-r--r-- | MulConstant.lua | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/MulConstant.lua b/MulConstant.lua new file mode 100644 index 0000000..982ab41 --- /dev/null +++ b/MulConstant.lua @@ -0,0 +1,21 @@ +local MulConstant, parent = torch.class('nn.MulConstant', 'nn.Module') + +function MulConstant:__init(constant_scalar) + parent.__init(self) + assert(type(constant_scalar) == 'number', 'input is not scalar!') + self.constant_scalar = constant_scalar +end + +function MulConstant:updateOutput(input) + self.output:resizeAs(input) + self.output:copy(input) + self.output:mul(self.constant_scalar) + return self.output +end + +function MulConstant:updateGradInput(input, gradOutput) + self.gradInput:resizeAs(gradOutput) + self.gradInput:copy(gradOutput) + self.gradInput:mul(self.constant_scalar) + return self.gradInput +end |