dgl.block_to_graph

dgl.block_to_graph(block)[源码]

将消息流图（MFG）作为一个 DGLBlock 对象转换为一个 DGLGraph。

DGL 会将所有源节点类型通过后缀 _src 进行重命名，并将所有目标节点类型通过后缀 _dst 进行重命名。

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

参数:: block (DGLBlock) – 消息流图（MFG）。
返回值:: 该图。
返回值类型:: 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')])