dgl.block_to_graph
- dgl.block_to_graph(block)[source]
将消息流图(MFG)作为
DGLBlock
对象转换为DGLGraph
。DGL将通过添加后缀
_src
来重命名所有源节点类型,并通过添加后缀_dst
来重命名所有目标节点类型。返回的图表上的特征将被保留。
- Parameters:
block (DGLBlock) – MFG。
- Returns:
图表。
- Return type:
示例
>>> block = dgl.create_block({ ... ('A', 'AB', 'B'): ([1, 2, 3], [2, 1, 0]), ... ('B', 'BA', 'A'): ([2, 1], [2, 3])}) >>> g = dgl.block_to_graph(block) >>> g Graph(num_nodes={'A_src': 4, 'B_src': 3, 'A_dst': 4, 'B_dst': 3}, num_edges={('A_src', 'AB', 'B_dst'): 3, ('B_src', 'BA', 'A_dst'): 2}, metagraph=[('A_src', 'B_dst', 'AB'), ('B_src', 'A_dst', 'BA')])