I have a graph dataset with 500 small graphs(each graph have about 5 nodes). A graph neural network is trained by the dataset for link prediction task. However, for an input such as (3 nodes), it can only get a fixed prediction result. How can I get multiple different predictions? Some details you can see in Fig below. Thanks for your reply!