import torch
import torch.nn as nn
import torch.nn.functional as F
 
class GCN(nn.Module):
    def __init__(self, num_classes, num_filters, dropout_rate=0.5):
        super(GCN, self).__init__()
        self.conv1 = nn.Conv2d(1, num_filters, (7, 7), 1, (3, 3))
        self.conv2 = nn.Conv2d(num_filters, num_classes, (1, 1), 1)
        self.dropout = nn.Dropout(dropout_rate)
 
    def forward(self, x):
        x = self.conv1(x)
        x = F.relu(x)
        x = self.dropout(x)
        x = self.conv2(x)
        x = F.log_softmax(x, dim=1)
        return x
 
# 示例输入
num_classes = 10  # 假设有10个类别
num_filters = 32  # 过滤器数量
dropout_rate = 0.5  #  dropout比例
input_data = torch.randn(1, 1, 28, 28)  # MNIST数据集的输入,假设使用28x28的图像
 
# 实例化GCN模型
model = GCN(num_classes, num_filters, dropout_rate)
output = model(input_data)
print(output.shape)  # 输出的形状
这段代码定义了一个基本的图卷积神经网络(GCN),它接受图像数据作为输入,并输出对应的分类概率。代码使用PyTorch框架,并假设输入数据是MNIST手写数字数据集的28x28图像。这个简单的例子展示了如何实现一个基本的图卷积层和全连接层,并使用ReLU激活函数和Dropout来防止过拟合。