import dgl
import torch
# 假设我们创建一个简单的图
# 创建图
g = dgl.graph(([0, 1, 2], [1, 2, 3]))
# 为节点添加特征,比如每个节点有2个特征
g.ndata['feat'] = torch.tensor([[1, 2], [3, 4], [5, 6], [7, 8]])
# 为边添加特征,比如每条边有1个特征
g.edata['weight'] = torch.tensor([0.1, 0.2, 0.3])
# 打印节点特征
print("Node features")
print(g.ndata)
# 打印边特征
print("Edge features")
print(g.edata)
print("Node features")
g.ndata['feat1']=torch.tensor([[3],[4],[5],[6]])
print(g.ndata)
dgl中的dgl.DGLGraph.ndata和dgl.DGLGraph.edata
于 2024-03-03 09:23:34 首次发布