Pytorch Graph Neural Network Decoder