通过访问data对象的属性,我如何知道哪个节点特性属于哪个节点?如果我正确理解它,data.x将保存节点特性。通过在下面运行for循环,我可以访问这些特性,但是我如何知道,如果它属于节点0或节点9的例子呢?
from torch_geometric.data import Data
edge_index = torch.tensor([[0, 1, 1, 2, 1, 9],
[1, 0, 2, 1, 8, 1]
], dtype=torch.long)
x = torch.tensor([[-5,7], [0,5], [0,9], [10,9]], dtype=torch.float)
data = Data(x=x, edge_index=edge_index)
for item in range(0, data.x.shape[0]):
print(item, data.x[item], data.edge_index.t()[item])https://stackoverflow.com/questions/71441535
复制相似问题