src/transformers/models/electra/modeling_electra.py/ElectraDiscriminatorPredictions/__init__
class ElectraDiscriminatorPredictions:    def __init__(self, config):
        super().__init__()

        self.dense = nn.Linear(config.hidden_size, config.hidden_size)
        self.dense_prediction = nn.Linear(config.hidden_size, 1)
        self.config = config