src/transformers/models/deit/modeling_deit.py/DeiTSelfOutput/forward
class DeiTSelfOutput:    def forward(self, hidden_states, input_tensor):

        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states)

        return hidden_states