Here is the code line, but I don't understand it. https://github.com/Franpin/TopoLogic/blob/2e1eab6cd91d2f5f9e4f066c60d65cef95d3166b/projects/topologic/models/dense_heads/relationship_head.py#L88