dgl.from_cugraph

dgl.from_cugraph(cugraph_graph)[source]

从 cugraph.Graph 对象创建图。

参数:

cugraph_graph (cugraph.Graph) –

持有图结构的 cugraph 图对象。节点和边的属性将被丢弃。

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

返回值:

创建的图。

返回类型:

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’))