class Net(nn.Module):

    def _init_(self): #神经网络结构，输入数据 1*32*32

        super(Net,self)._init_()

        #conve

        self.conv1=nn.Conv2d(1,6,3) #输入配道1，输出判道6，卷积3*3

        #第2层

        self.conv2=nn.Conv2d(6,16,3)#输入配道6，输出判道16，卷积3*3

        #full connect

        self.fc1=nn.Linear(16*28*28,512)

        self.fc2=nn.Linear(512,64)

        self.fc3=nn.Linear(64,2)

    def forward(self,x):

        x = self.conv1(x)

        x= F.relu(x)

        x=self.conv2(x)

        x=F.relu(x)

        x= x.view(-1,16*28*28)

        x=self.fc1(x)

        x=F.relu(x)

        x=self.fc2(x)

        x=F.relu(x)

        x=self.fc3(x)

        return x