本文共 518 字，大约阅读时间需要 1 分钟。

2019.11.30

from d2l import torch as d2limport torchfrom torch import nnclass Reshape(torch.nn.Module):    def forward(self, x):        return x.view(-1, 1, 28, 28)net = torch.nn.Sequential(    Reshape(),    nn.Conv2d(1, 6, kernel_size=5, padding=2), nn.Sigmoid(),    nn.AvgPool2d(kernel_size=2, stride=2),    nn.Conv2d(6, 16, kernel_size=5), nn.Sigmoid(),    nn.AvgPool2d(kernel_size=2, stride=2),    nn.Flatten(),    nn.Linear(16 * 5 * 5, 120), nn.Sigmoid(),    nn.Linear(120, 84), nn.Sigmoid(),    nn.Linear(84, 10))

转载地址：http://hrzai.baihongyu.com/