I am trying to train a graph neural network using pytorch, but the problem is the final output contain the same values for each row, here is the layer of my network