在传入的时候,我定义的BATCH_SIZE为512,所以具体的输入维度为(512,1,28,28)
我的CNN卷积神经网络的为两层卷积层,两次激活函数,两层池化层,和两层全连接层
卷积核设为5X5,步长...(BATCH_SIZE,1,28,28)
1.2经过激活函数,只将张量中的为负数的值变为0,不改变shape,各维度不变(BATCH_SIZE,14,28,28)
1.3经过最大池化层,将图片缩小,降采样...(BATCH_SIZE,28,7,7)
3.利用view函数,将张量拉平,shape变为(BATCH_SIZE,2877)
4.1经过第一层全连接层,将(2877)变为200,高度提纯,一个全连接层将卷积层提取的特征进行线性组合...__init__()
self.conv1 = nn.Conv2d(1, 14, 5,1,2) # 卷积函数 1:灰度图片的通道 14:输出通达 5:kernel...# 输入:batch*14*14*14,输出:batch*28*7*7
x = x.view(input_size, -1) # 拉平,-1自动计算维度,28*7*7=1372