diff options
Diffstat (limited to 'contrib/lua-torch/nn/Exp.lua')
| -rw-r--r-- | contrib/lua-torch/nn/Exp.lua | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/contrib/lua-torch/nn/Exp.lua b/contrib/lua-torch/nn/Exp.lua new file mode 100644 index 000000000..f41569026 --- /dev/null +++ b/contrib/lua-torch/nn/Exp.lua @@ -0,0 +1,9 @@ +local Exp = torch.class('nn.Exp', 'nn.Module') + +function Exp:updateOutput(input) + return self.output:exp(input) +end + +function Exp:updateGradInput(input, gradOutput) + return self.gradInput:cmul(self.output, gradOutput) +end |