Add functional conversion of HardShrink

1 files changed, 11 insertions, 2 deletions

diff --git a/HardShrink.lua b/HardShrink.lua
index 7dfeaca..85ff590 100644
--- a/HardShrink.lua
+++ b/HardShrink.lua
@@ -6,11 +6,20 @@ function HardShrink:__init(lam)
 end
 
 function HardShrink:updateOutput(input)
-   input.nn.HardShrink_updateOutput(self, input)
+   input.THNN.HardShrink_updateOutput(
+      input:cdata(),
+      self.output:cdata(),
+      self.lambda
+   )
    return self.output
 end
 
 function HardShrink:updateGradInput(input, gradOutput)
-   input.nn.HardShrink_updateGradInput(self, input, gradOutput)
+   input.THNN.HardShrink_updateGradInput(
+      input:cdata(),
+      gradOutput:cdata(),
+      self.gradInput:cdata(),
+      self.lambda
+   )
    return self.gradInput
 end