构造一个3层全连接层的模型的代码 python

下面是一个简单的 3 层全连接层模型的 Python 代码示例：

import torch.nn as nn

class ThreeLayerFC(nn.Module):
    def __init__(self, input_dim, hidden_dim, output_dim):
        super(ThreeLayerFC, self).__init__()
        self.layer1 = nn.Linear(input_dim, hidden_dim)
        self.layer2 = nn.Linear(hidden_dim, hidden_dim)
        self.layer3 = nn.Linear(hidden_dim, output_dim)
        
    def forward(self, x):
        x = nn.functional.relu(self.layer1(x))
        x = nn.functional.relu(self.layer2(x))
        x = self.layer3(x)
        return x

这个模型包含三个全连接层，每个层之间都使用 ReLU 激活函数。输入层的维度为 `input_dim`，第一个和第二个隐藏层的维度都是 `hidden_dim`，输出层的维度为 `output_dim`。在前向传播中，输入先经过第一个全连接层，然后是第二个全连接层，最后是输出层。