dgl.DGLGraph.device

property DGLGraph.device

获取图形的设备。

Returns:

图形的设备，应该是一个特定框架的设备对象 （例如，torch.device）。

Return type:

设备上下文

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import torch

创建一个同构图用于演示。

>>> g = dgl.graph((torch.tensor([0, 1]), torch.tensor([1, 2])))
>>> print(g.device)
device(type='cpu')

异构图的情况也是如此。