diagram-projectTGNs with TorchGeometric