src/transformers/models/beit/modeling_beit.py/BeitOutput/forward
class BeitOutput:    def forward(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states)

        return hidden_states