)
在前一节中,我们了解了如何使用PyTorch的unsqueeze()方法添加批处理维度来传递单个图像。...彩色通道的数量从1个增加到6个。在我们通过了第一个卷积层之后,我们不再认为通道是彩色通道。我们只是把它们当做输出通道。...我们可以通过检查权重张量的形状来看到这一点:
> self.conv1.weight.shape
torch.Size([6, 1, 5, 5])
张量的形状由下式给出:
(number of filters...最大池化操作
池化操作通过从张量中的每个2x2位置提取最大值来进一步减小张量的形状。...([[ 0.1009, -0.0842, 0.0349, -0.0640, 0.0754, -0.0057, 0.0878, 0.0296, 0.0345, 0.0236]])
下表总结了形状更改操作以及每种形状的结果形状