代码练习系列（二）——搭建一个可用于手写数字识别的卷积网络模型（黑白图像，28*28大小，10个分类）

import torch
from torch import nn

class Model(nn.Module):

    def __init__(self):
        super(Model, self).__init__()
        self.conv1 = nn.Conv2d(in_channels=1, 
                               out_channels=16,
                               kernel_size=3,
                               stride=1,
                               padding=1)
        self.bn1 = nn.BatchNorm2d(num_features=16)
        self.mp1 = nn.MaxPool2d(kernel_size=2, 
                                stride=2, 
                                padding=0)
        self.conv2 = nn.Conv2d(in_channels=16, 
                               out_channels=64,
                               kernel_size=3,