src/transformers/models/electra/modeling_electra.py/ElectraDiscriminatorPredictions/forward
class ElectraDiscriminatorPredictions:    def forward(self, discriminator_hidden_states):
        hidden_states = self.dense(discriminator_hidden_states)
        hidden_states = get_activation(self.config.hidden_act)(hidden_states)
        logits = self.dense_prediction(hidden_states).squeeze(-1)

        return logits