dgl.DGLGraph.ntypes

property DGLGraph.ntypes

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

返回:

一个列表，包含所有节点类型的名称。

返回类型:

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']