dgl 给图打标签(不是点、也不是边)
时间: 2023-08-06 19:03:02 浏览: 151
为图片添加标签
DGL可以通过给节点或边添加属性来给图打标签。如果需要为整个图打标签,可以给图对象添加属性。以下是一个示例:
```python
import dgl
import torch
# 创建一个包含3个节点和2条边的图
g = dgl.graph(([0, 1], [1, 2]))
# 为图对象添加标签
g.graph['label'] = torch.tensor([1])
# 为节点添加标签
g.ndata['label'] = torch.tensor([0, 1, 0])
# 为边添加标签
g.edata['label'] = torch.tensor([2, 3])
```
在上面的示例中,我们为图对象添加了一个名为'label'的标签。我们还为节点添加了名为'label'的标签,其中第一个和第三个节点的标签为0,第二个节点的标签为1。我们还为边添加了名为'label'的标签,其中第一条边的标签为2,第二条边的标签为3。
阅读全文