dgl.DGLGraph.srctypes

property DGLGraph.srctypes

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

如果图可以将其节点类型进一步划分为两个子集 A 和 B，其中所有边都从 A 中的节点类型指向 B 中的节点类型，我们将此图称为 单部二分图，A 中的节点称为 源 节点，B 中的节点称为 目标 节点。如果图不是单部二分图，则源节点和目标节点就是图中的整个节点集合。

返回值:

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

返回类型:

list[str]

另请参阅

dsttypes, is_unibipartite

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

查询单部二分图。

>>> g = dgl.heterograph({
...     ('user', 'plays', 'game'): (torch.tensor([0]), torch.tensor([1])),
...     ('developer', 'develops', 'game'): (torch.tensor([1]), torch.tensor([2]))
... })
>>> g.srctypes
['developer', 'user']

查询非单部二分图。

>>> g = dgl.heterograph({
...     ('user', 'follows', 'user'): (torch.tensor([0]), torch.tensor([1])),
...     ('developer', 'develops', 'game'): (torch.tensor([1]), torch.tensor([2]))
... })
>>> g.srctypes
['developer', 'game', 'user']