Graph Neural Network Pytorch