src/transformers/models/levit/modeling_levit.py/LevitClassificationLayer/forward
class LevitClassificationLayer:    def forward(self, hidden_state):
        hidden_state = self.batch_norm(hidden_state)
        logits = self.linear(hidden_state)
        return logits