weixin_52993324的博客

原创 多层全连接网络实现MNIST手写数字分类

self.layer1 = nn.Linear(in_dim, n_hidden_1) # 第一个全连接层，将输入纬度(in_dim)转换为第一隐藏层纬度(n_hidden_1)img = img.view(img.size(0), -1) # 将图片从(64（batch_size）,1,28,28)转换为（64，28*28）out = model(img) # 等价于 out = model.forward(img)nn.BatchNorm1d(n_hidden_2), # 批归一化层。