dgl.DGLGraph.ntypes

property DGLGraph.ntypes

返回图中所有节点类型的名称。

Returns:

列表中所有节点类型的名称。

Return type:

list[str]

注释

DGL 内部为每种节点类型分配一个整数 ID。返回的节点类型名称根据其 ID 排序。

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import torch

>>> g = dgl.heterograph({
...     ('user', 'follows', 'user'): (torch.tensor([0, 1]), torch.tensor([1, 2])),
...     ('user', 'follows', 'game'): (torch.tensor([0, 1, 2]), torch.tensor([1, 2, 3])),
...     ('user', 'plays', 'game'): (torch.tensor([1, 3]), torch.tensor([2, 3]))
... })
>>> g.ntypes
['game', 'user']