| layer | layer setting |
|---|---|
| F.relu(self.conv1(x)) | self.conv1 = nn.Conv2d(1, 6, 5) |
| F.max pool2d(x, 2, 2) | |
| F.relu(self.conv2(x)) | self.conv2 = nn.Conv2d(6, 16, 5) |
| x.view(−1, 16*4) | |
| F.relu(self.fc1(x)) | self.fc1 = nn.Linear(16*4*4, 120) |
| x= F.relu(self.fc2(x)) | self.fc2 = nn.Linear(120, 84) |
| x = self.fc3(x) | self.fc3 = nn.Linear(84, 10) |
| F.log softmax(x, dim=1) |
Official websites use .gov
A
.gov website belongs to an official
government organization in the United States.
Secure .gov websites use HTTPS
A lock (
) or https:// means you've safely
connected to the .gov website. Share sensitive
information only on official, secure websites.