From e406feabe2b1bd7f6b7a15826aff1e925fe713d6 Mon Sep 17 00:00:00 2001
From: Jonathan Tompson <tompson@cims.nyu.edu>
Date: Wed, 25 Jun 2014 12:14:12 -0400
Subject: Added very simple add and mul constant modules.

---
 MulConstant.lua | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 MulConstant.lua

(limited to 'MulConstant.lua')

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
-- 
cgit v1.2.3