GNN Policy Code Functions I use for RL agents based on graph neural networks. These days only functional.py is used.