Implementing a call operator for nn modules and criterions.

1 files changed, 11 insertions, 1 deletions

diff --git a/Module.lua b/Module.lua
index 98ee86c..dd61dc6 100644
--- a/Module.lua
+++ b/Module.lua
@@ -22,7 +22,7 @@ function Module:updateOutput(input)
 end
 
 function Module:forward(input)
-   return self:updateOutput(input, target)
+   return self:updateOutput(input)
 end
 
 function Module:backward(input, gradOutput)
@@ -212,3 +212,13 @@ function Module:getParameters()
    -- return new flat vector that contains all discrete parameters
    return flatParameters, flatGradParameters
 end
+
+function Module:__call__(input, gradOutput)
+   self:forward(input)
+   if gradOutput then
+      self:backward(input, gradOutput)
+      return self.output, self.gradInput
+   else
+      return self.output
+   end
+end