dgl.from_cugraph

dgl.from_cugraph(cugraph_graph)[source]

从cugraph.Graph对象创建图形。

Parameters:

cugraph_graph (cugraph.Graph) –

cugraph 图对象，用于保存图结构。节点和边的属性被丢弃。

如果输入图是无向的，DGL 会通过 cugraph.Graph.to_directed() 将其转换为有向图。

Returns:

创建的图表。

Return type:

DGLGraph

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import cugraph
>>> import cudf

创建一个cugraph图。 >>> cugraph_g = cugraph.Graph(directed=True) >>> df = cudf.DataFrame({“source”:[0, 1, 2, 3],

“destination”:[1, 2, 3, 0]})

>>> cugraph_g.from_cudf_edgelist(df)

将其转换为DGLGraph >>> g = dgl.from_cugraph(cugraph_g) >>> g.edges() (tensor([1, 2, 3, 0], device=’cuda:0’), tensor([2, 3, 0, 1], device=’cuda:0’))