python - Pytorch 几何：如何解释以下代码片段中的输入？

Question

我正在阅读https://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.html上的 PyTorch 几何文档

在此页面上，有一个代码片段：

import torch
from torch_geometric.data import Data

edge_index = torch.tensor([[0, 1, 1, 2],
                           [1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)

data = Data(x=x, edge_index=edge_index)

上述代码片段最后一行的输出是：

Data(edge_index=[2, 4], x=[3, 1])

edge_index2和4怎么样？如果我理解正确，则定义了四个边，索引从 0 开始。这个假设是错误的吗？还有，什么x =[3, 1]意思？

Data是一个类，所以我不会期望它返回任何东西。类定义在这里：https ://pytorch-geometric.readthedocs.io/en/latest/modules/data.html 。我阅读了文档。x应该是节点特征矩阵并且edge_index应该是图连通性。但我不明白我在 jupyter 笔记本中交叉检查的控制台输出。

Answer 1

好的，我想我已经了解了 output Data(edge_index=[2, 4], x=[3, 1])。这里 [2,4] 是 的维度，edge_index[3,1] 是 的维度x。但是，如果我错了，请任何人纠正我。