summaryrefslogtreecommitdiffstats
path: root/contrib/lua-torch/nn/Exp.lua
blob: f41569026924aa16098b83200f25f28038bfe636 (plain)
1
2
3
4
5
6
7
8
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