用途一：數據集太大，無法畫圖，取子圖看看是有向圖/無向圖

import dgl import matplotlib.pyplot as plt import networkx as nx G = dgl.DGLGraph() G.add_nodes(5) # G.add_edges([0, 1, 2, 3, 4], [1, 2, 3, 4, 0]) # 有向圖 G.add_edges([0, 1, 2, 3, 4, 1, 2, 3, 4, 0], [1, 2, 3, 4, 0, 0, 1, 2, 3, 4]) # 無向圖（雙向圖） SG = G.subgraph([0, 1, 4]) SG.nodes() # tensor([0, 1, 2]) SG.edges() # (tensor([0, 2]), tensor([1, 0])) SG.parent_nid # tensor([0, 1, 4]) SG.parent_eid # tensor([0, 4])nx_G1 = G.to_networkx() nx_G2 = SG.to_networkx() plt.subplot(121) nx.draw_shell(nx_G1, with_labels=True, font_weight='bold') # 節點按序排列 plt.subplot(122) nx.draw_shell(nx_G2, with_labels=True, font_weight='bold') # 節點按序排列 plt.show()

有向圖的子圖是有向圖
無向圖的子圖是無向圖

創作挑戰賽新人創作獎勵來咯，堅持創作打卡瓜分現金大獎

總結

以上是生活随笔為你收集整理的DGL_子图的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。