I am trying to represent a large graph in PyTorch by placing each element in it\'s own layer and then connecting layers. I would then like to find communities in that graph usin