Visualize the structures of GCNModel and GATModel

Hi, I am using GCNModel and GATModel. May I ask, is there a way to summarize and visualize the structures of these models such as plot_model in Keras, please? Thanks a lot!

I believe we don’t have a good way to do this right now, but this could be a great contribution for a new contributor!

Thank you. I will try.