tf.keras.layers.Flatten

概述

将输入的多维数组展平

神经网络的输入层是一个 一维向量,所以需要把输入的多维数组展平

比如:把输入的图像矩阵(二维数组)展平,从 28 x 28 变为 1 维数组,长度是 784

初始化

__init__(
)
参数
  • input_shape:输入数据的形状

感谢:
https://www.jianshu.com/p/ff48a508d490


原文出处:/show_1EF48veCzsfa.html