开发者社区 问答 正文

如何在PyTorch中实现多层神经网络?

如何在PyTorch中实现多层神经网络?

展开
收起
萝卜丝丸子 2024-06-03 20:10:48 64 分享 版权
1 条回答
写回答
取消 提交回答
  • 在PyTorch中实现多层神经网络可以通过在模型中添加多个nn.Linear层,并在forward函数中依次传递输入数据。例如,一个包含两个隐藏层的神经网络可以这样定义:

    class MultiLayerNet(nn.Module):  
        def __init__(self, n_feature, n_hidden1, n_hidden2, n_output):  
            super(MultiLayerNet, self).__init__()  
            self.layer1 = nn.Linear(n_feature, n_hidden1)  
            self.relu = nn.ReLU()  
            self.layer2 = nn.Linear(n_hidden1, n_hidden2)  
            self.output = nn.Linear(n_hidden2, n_output)  
    
        def forward(self, x):  
            x = self.relu(self.layer1(x))  
            x = self.relu(self.layer2(x))  
            y = self.output(x)  
            return y
    
    2024-06-03 22:20:59
    赞同 1 展开评论