python - 卷积神经网络中的 Flatten() 和 Dense() 层有什么区别？

Question

我有他们之间的严重怀疑。任何人都可以用例子和一些想法详细说明。

标签： pythonmachine-learningneural-networkdeep-learningconv-neural-network

Answer 1

顾名思义，Flatten 将您的多维矩阵 (Batch.Size x Img.W x Img.H x Kernel.Size) 转换为一个不错的单个二维矩阵：(Batch.Size x (Img.W x Img.H x内核大小））。在反向传播期间，它还将您的大小增量 (Batch.Size x (Img.W x Img.H x Kernel.Size)) 转换回原始 (Batch.Size x Img.W x Img.H x Kernel.Size)。

密集层当然是标准的全连接层。