PyTorch卷积模块

包括卷积层、池化层、提取层结构 构造一个简单的多层卷积神经网络 import numpy as np import torch from torch.autograd import Variable from torch import nn, optim import matplotlib.pyplot as plt class SimpleCNN(nn.Module): def __init__(self): super(SimpleCNN, self).__init__() #b, 3, 32, 32 layer1 = nn.Sequential