dgl.DGLGraph.node_attr_schemes

DGLGraph.node_attr_schemes(ntype=None)

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

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

Parameters:

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

Returns:

一个将特征名称映射到其相关特征方案的字典。

Return type:

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