diff options
Diffstat (limited to 'Sqrt.lua')
-rw-r--r-- | Sqrt.lua | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/Sqrt.lua b/Sqrt.lua new file mode 100644 index 0000000..664d434 --- /dev/null +++ b/Sqrt.lua @@ -0,0 +1,13 @@ +local Sqrt, parent = torch.class('nn.Sqrt','nn.Module') + +function Sqrt:__init(args) + parent.__init(self) +end + +function Sqrt:updateOutput(input) + return input.nn.Sqrt_updateOutput(self,input) +end + +function Sqrt:updateGradInput(input, gradOutput) + return input.nn.Sqrt_updateGradInput(self,input,gradOutput) +end |