- 博客(1)
- 收藏
- 关注
原创 多层全连接网络实现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), # 批归一化层。
2024-06-19 15:47:31
515
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人