src/transformers/models/audio_spectrogram_transformer/modeling_audio_spectrogram_transformer.py/ASTMLPHead/forward
class ASTMLPHead:    def forward(self, hidden_state):
        hidden_state = self.layernorm(hidden_state)
        hidden_state = self.dense(hidden_state)
        return hidden_state