PyTorch学习快速搭建网络

Posted 2021-06-07 My heart will go ~~

本节主要介绍快速搭建神经网络
通过对比两个网络的定义，快速搭建网络代码十分简洁。

import torch
import torch.nn.functional as F

class Net(torch.nn.Module):
    def __init__(self,n_feature,n_hidden,n_output):
        super(Net,self).__init__()
        self.hidden=torch.nn.Linear(n_feature,n_hidden)
        self.out=torch.nn.Linear(n_hidden,n_output)
        
    def forward(self,x):
        x=F.relu(self.hidden(x))
        x=self.out(x)
        return x

net1=Net(2,10,2)
print(net1)

#第二种方法定义网络（快速定义网络

net2 = torch.nn.Sequential(#累神经层就可以了
    torch.nn.Linear(2, 10),
    torch.nn.ReLU(),
    torch.nn.Linear(10, 2)
)
print(net2)

那搭建的网络有何不同呢？我们编译之后查看网络就可以了。