src/transformers/models/big_bird/modeling_big_bird.py/BigBirdPredictionHeadTransform/forward
class BigBirdPredictionHeadTransform:    def forward(self, hidden_states):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.transform_act_fn(hidden_states)
        hidden_states = self.LayerNorm(hidden_states)
        return hidden_states