src/transformers/models/segformer/modeling_segformer.py/SegformerMLP/forward
class SegformerMLP:    def forward(self, hidden_states: torch.Tensor):
        hidden_states = hidden_states.flatten(2).transpose(1, 2)
        hidden_states = self.proj(hidden_states)
        return hidden_states