图神经网络GNN：创建图和展示图

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

''' 摘自https://docs.dgl.ai/en/0.6.x/guide_cn/graph-graphs-nodes-edges.html '''   import dgl import torch as th import networkx as nx import matplotlib.pyplot as plt   # 边 0->1, 0->2, 0->3, 1->3 u, v = th.tensor([0, 0, 0, 1]), th.tensor([1, 2, 3, 3]) graph = dgl.graph((u, v)) fig, ax = plt.subplots() nx.draw(graph.to_networkx(), ax=ax)   # 将图转为networkx形式 ax.set_title('Graph') plt.show()

　　

结果：