class Model(Module): def __init__(self): super(Model, self).__init__() self.conv1 = nn.Conv2d(1, 6, 5) self.relu1 = nn.ReLU() self.pool1 = nn.MaxPool2d(2) self.conv2 = nn.Conv2d(6, 16, 5) self.relu2 = nn.ReLU() self.pool2 = nn.MaxPool2d(2) self.fc1 = nn.Linear(256, 120) self.relu3 = nn.ReLU() self.fc2 = nn.Linear(120, 84) self.relu4 = nn.ReLU() self.fc3 = nn.Linear(84, 10) self.relu5 = nn.ReLU() def forward(self, x): y = self.conv1(x) y = self.relu1(y) y = self.pool1(y) y = self.conv2(y) y = self.relu2(y) y = self.pool2(y) y = y.view(y.shape[0], -1) y = self.fc1(y) y = self.relu3(y) y = self.fc2(y) y = self.relu4(y) y = self.fc3(y) y = self.relu5(y) return y

这一个简单的卷积神经网络模型的定义，用于图像分类任务。该模型包含了两个卷积层（`self.conv1` 和 `self.conv2`）、两个池化层（`self.pool1` 和 `self.pool2`）以及三个全连接层（`self.fc1`、`self.fc2` 和 `self.fc3`）。在每个卷积层和全连接层之间，都使用了 ReLU 激活函数（`self.relu1`、`self.relu2`、`self.relu3`、`self.relu4` 和 `self.relu5`）。

在 `forward()` 方法中，输入数据 `x` 会通过网络的不同层进行前向传播操作。首先，输入通过第一个卷积层、ReLU 激活函数和池化层。然后，结果再经过第二个卷积层、ReLU 激活函数和池化层。接下来，通过将特征图展平为一维向量（使用 `y.view(y.shape[0], -1)`），将结果传递给第一个全连接层、ReLU 激活函数、第二个全连接层、ReLU 激活函数以及最后一个全连接层。最后，通过 ReLU 激活函数输出分类结果。

你可以根据自己的需求修改和扩展这个模型。注意，这个模型适用于输入通道数为 1 的灰度图像，如果你的数据是彩色图像或者通道数不是 1，你需要相应地修改卷积层的输入通道数。