    def clamp_param(self):
        """this need to be called after each update"""
        self.current_val.data.clamp_(0, 1)
this need to be called after each update
examples/adaptive_span/adaptive_span_attention.py