src/transformers/models/funnel/modeling_funnel.py/FunnelDiscriminatorPredictions/__init__
class FunnelDiscriminatorPredictions:    def __init__(self, config: FunnelConfig) -> None:
        super().__init__()
        self.config = config
        self.dense = nn.Linear(config.d_model, config.d_model)
        self.dense_prediction = nn.Linear(config.d_model, 1)