PyTorch Geometric GCN 模型图像特征提取：输入形状解释

在使用 PyTorch Geometric 中的 GCN 模型进行图像特征提取时，输入 GCN 网络的 x 形状往往是 torch.Size([1, 4, 1, 1])。这是由于在定义 CNN 模型时，经过两次卷积操作后，图像的空间维度被压缩为 1x1。

以下是该形状的具体含义：

• 1: 表示 batch_size，即每个 batch 中的样本数量。
• 4: 表示特征通道数，即 CNN 模型输出的特征数量。
• 1x1: 表示空间维度。经过两次卷积操作，图像的空间信息已经被压缩为 1x1。

由于 GCN 模型使用 CNN 模型提取的图像特征作为输入，因此输入的形状是 torch.Size([1, 4, 1, 1])。

简而言之，torch.Size([1, 4, 1, 1]) 表示一个 batch 中的图像经过 CNN 模型提取特征后，得到一个 4 通道的 1x1 特征矩阵。这个矩阵会被作为 GCN 模型的输入，以便进行后续的图卷积操作。