examples/research_projects/movement-pruning/emmental/modeling_bert_masked.py/BertSelfOutput/forward
class BertSelfOutput:    def forward(self, hidden_states, input_tensor, threshold):
        hidden_states = self.dense(hidden_states, threshold=threshold)
        hidden_states = self.dropout(hidden_states)
        hidden_states = self.LayerNorm(hidden_states + input_tensor)
        return hidden_states