Porting tf-gnn to DL4J

For completeness sake, here’s the answer to the question, in another thread: