\'\'\' 摘自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.savefig(\'fig_graph.png\')
结果：