src/transformers/models/albert/modeling_flax_albert.py/FlaxAlbertSOPHead/__call__
class FlaxAlbertSOPHead:    def __call__(self, pooled_output, deterministic=True):
        pooled_output = self.dropout(pooled_output, deterministic=deterministic)
        logits = self.classifier(pooled_output)
        return logits