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
|