dgl.DGLGraph.idtype

property DGLGraph.idtype

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

Returns:

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

Return type:

特定框架的设备对象

示例

以下示例使用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