diff options
author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2018-05-23 18:14:15 +0100 |
---|---|---|
committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2018-05-23 18:14:15 +0100 |
commit | 714eb56e1760fdfb26afccde92664d3a2f1e8435 (patch) | |
tree | 84d1399acbb92f852b4bd64f9ea5412680b0c6ab /contrib/lua-torch/nn/Sqrt.lua | |
parent | 220a51ff68013dd668a45b78c60a7b8bfc10f074 (diff) | |
download | rspamd-714eb56e1760fdfb26afccde92664d3a2f1e8435.tar.gz rspamd-714eb56e1760fdfb26afccde92664d3a2f1e8435.zip |
[Minor] Move lua contrib libraries to lua- prefix
Diffstat (limited to 'contrib/lua-torch/nn/Sqrt.lua')
-rw-r--r-- | contrib/lua-torch/nn/Sqrt.lua | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/contrib/lua-torch/nn/Sqrt.lua b/contrib/lua-torch/nn/Sqrt.lua new file mode 100644 index 000000000..df354a175 --- /dev/null +++ b/contrib/lua-torch/nn/Sqrt.lua @@ -0,0 +1,26 @@ +local Sqrt, parent = torch.class('nn.Sqrt','nn.Module') + +function Sqrt:__init(b) + parent.__init(self) + self.eps = b or 0 +end + +function Sqrt:updateOutput(input) + self.eps = self.eps or 0 + input.THNN.Sqrt_updateOutput( + input:cdata(), + self.output:cdata(), + self.eps + ) + return self.output +end + +function Sqrt:updateGradInput(input, gradOutput) + input.THNN.Sqrt_updateGradInput( + input:cdata(), + gradOutput:cdata(), + self.gradInput:cdata(), + self.output:cdata() + ) + return self.gradInput +end |