dgl.DGLGraph.device

property DGLGraph.device

获取图的设备。

返回:

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

返回类型:

设备上下文

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import torch

创建一个同构图进行演示。

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

异构图的情况也是一样的。