src/transformers/models/mobilebert/modeling_mobilebert.py/MobileBertOnlyNSPHead/forward
class MobileBertOnlyNSPHead:    def forward(self, pooled_output: torch.Tensor) -> torch.Tensor:
        seq_relationship_score = self.seq_relationship(pooled_output)
        return seq_relationship_score