I am using Pytorch-Geometric library to implement a Graph Neural Network for a regression problem. The model is defined as:
<