diff options
Diffstat (limited to 'Threshold.lua')
| -rw-r--r-- | Threshold.lua | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/Threshold.lua b/Threshold.lua new file mode 100644 index 0000000..6083957 --- /dev/null +++ b/Threshold.lua @@ -0,0 +1,20 @@ +local Threshold, parent = torch.class('nn.Threshold','nn.Module') + +function Threshold:__init(th,v) + parent.__init(self) + self.threshold = th or 1e-6 + self.val = v or 0 + if (th and type(th) ~= 'number') or (v and type(v) ~= 'number') then + error('nn.Threshold(threshold, value)') + end +end + +function Threshold:updateOutput(input) + input.nn.Threshold_updateOutput(self, input) + return self.output +end + +function Threshold:updateGradInput(input, gradOutput) + input.nn.Threshold_updateGradInput(self, input, gradOutput) + return self.gradInput +end |