diff options
Diffstat (limited to 'Minus.lua')
-rw-r--r-- | Minus.lua | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -1,11 +1,11 @@ local Minus, parent = torch.class('nn.Minus', 'nn.Module') -function Minus:forward(input) +function Minus:updateOutput(input) self.output:resizeAs(input):copy(input):mul(-1) return self.output end -function Minus:backward(input, gradOutput) +function Minus:updateGradInput(input, gradOutput) self.gradInput:resizeAs(input):copy(gradOutput):mul(-1) return self.gradInput end |