dgl.DGLGraph.to_cugraph

DGLGraph.to_cugraph()

将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