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

? ? ? ??