src/transformers/models/fnet/modeling_fnet.py/FNetOnlyNSPHead/forward
class FNetOnlyNSPHead:    def forward(self, pooled_output):
        seq_relationship_score = self.seq_relationship(pooled_output)
        return seq_relationship_score