dgl.DGLGraph.node_attr_schemes

DGLGraph.node_attr_schemes(ntype=None)

返回指定节点类型的节点特征方案。

特征的方案描述了它的形状和数据类型。

参数：

ntype (str, 可选) – 节点类型名称。如果图中只有一种节点类型，则可以省略。

返回值：

返回一个字典，将特征名称映射到其关联的特征方案。

返回类型：

dict[str, Scheme]

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch

查询同构图。

>>> g = dgl.graph((torch.tensor([0, 1]), torch.tensor([1, 2])))
>>> g.ndata['h1'] = torch.randn(3, 1)
>>> g.ndata['h2'] = torch.randn(3, 2)
>>> g.node_attr_schemes()
{'h1': Scheme(shape=(1,), dtype=torch.float32),
 'h2': Scheme(shape=(2,), dtype=torch.float32)}

查询具有多种节点类型的异构图。

>>> g = dgl.heterograph({('user', 'plays', 'game'):
...                      (torch.tensor([1, 2]), torch.tensor([3, 4]))})
>>> g.nodes['user'].data['h1'] = torch.randn(3, 1)
>>> g.nodes['user'].data['h2'] = torch.randn(3, 2)
>>> g.node_attr_schemes('user')
{'h1': Scheme(shape=(1,), dtype=torch.float32),
 'h2': Scheme(shape=(2,), dtype=torch.float32)}

另请参阅

edge_attr_schemes