3*3卷积核与2*5卷积核对神经元大小的设置
#这里kerner_size = 2*5
class CONV_NET(torch.nn.Module): #CONV_NET类继承nn.Module类
def __init__(self):
super(CONV_NET, self).__init__() #使CONV_NET类包含父类nn.Module的所有属性
# super()需要两个实参,子类名和对象self
self.conv1 = nn.Conv2d(1, 32, (2, 5), 1, padding=0)
self.conv2 = nn.Conv2d(32, 1
1