src/transformers/models/albert/modeling_albert.py/AlbertSOPHead/forward
class AlbertSOPHead:    def forward(self, pooled_output: torch.Tensor) -> torch.Tensor:
        dropout_pooled_output = self.dropout(pooled_output)
        logits = self.classifier(dropout_pooled_output)
        return logits