张量扁平化——CNN的Flatten操作
张量扁平化操作是卷积神经网络中的一个常见操作。这是因为在全连接层接受输入之前,传递给全连接层的卷积层输出必须被扁平化。我们了解到卷积神经网络的张量输入通常有4个轴,一个用于批处理大小,一个用于颜色通道,还有一个用于高度和宽度即:(批量大小、通道、高度、宽度)那么现在我们以一张图片为例,看看如何将它扁平化这是一张彩色图片,也就是有r,g,b共三个通道,大小为128*128,按照上面的表示方法,我们可以将它表示为[1,3,128,128]code:好吧,可是看到,张量已经被拉平了,达到了我们预