dgl.DGLGraph.idtype

property DGLGraph.idtype

用于存储结构相关图信息（如节点和边 ID）的数据类型。

返回:

例如，对于 PyTorch，这可以是 torch.int32 或 torch.int64。

返回类型:

特定于框架的设备对象

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

>>> src_ids = torch.tensor([0, 0, 1])
>>> dst_ids = torch.tensor([1, 2, 2])
>>> g = dgl.graph((src_ids, dst_ids))
>>> g.idtype
torch.int64
>>> g = dgl.graph((src_ids, dst_ids), idtype=torch.int32)
>>> g.idtype
torch.int32

另请参阅

long、int