text: import torch
  from .Module import Module
  
  
  class HardShrink(Module):
  
      def __init__(self, lambd=0.5):
          assert type(lambd) == float
          super(HardShrink, self).__init__()
          self.lambd = lambd
  
      def updateOutput(self, input):
          self._backend.HardShrink_updateOutput(
              self._backend.library_state,
              input,
              self.output,
              self.lambd
          )
          return self.output
  
      def updateGradInput(self, input, gradOutput):
          self._backend.HardShrink_updateGradInput(
              self._backend.library_state,
              input,
              gradOutput,
              self.gradInput,
              self.lambd
          )
          return self.gradInput
  