dgl.to_cugraph

dgl.to_cugraph(g)

将 DGL 图转换为 cugraph.Graph 并返回。

参数:

g (DGLGraph) – 同构图。

返回:

转换后的 cugraph 图。

返回类型:

cugraph.Graph

注意

此函数仅支持 GPU 图输入。

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import cugraph
>>> import torch

>>> g = dgl.graph((torch.tensor([1, 2]), torch.tensor([1, 3]))).to('cuda')
>>> cugraph_g = g.to_cugraph()
>>> cugraph_g.edges()
    src  dst
0    2    3
1    1    1