dgl.block_to_graph

dgl.block_to_graph(block)[source]

将消息流图(MFG)作为DGLBlock对象转换为DGLGraph

DGL将通过添加后缀_src来重命名所有源节点类型,并通过添加后缀_dst来重命名所有目标节点类型。

返回的图表上的特征将被保留。

Parameters:

block (DGLBlock) – MFG。

Returns:

图表。

Return type:

DGLGraph

示例

>>> 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')])