目录
tf.keras.layers.Flatten的作用
可以将输入展开成一维的,不对batch size做展开
这个层一般用于将输入特征展开,便于后面接续全连接层。比如,输入数据都是20x20x1的灰度图片,现在想在这个图片后面直接连个全连接层。tf.keras.layers.Flatten就派上用场了,将每张图片展开成400的一维数组。
tf.keras.layers.Flatten不对batch size做展开,比如是如果有(batch_size,20,20)的输入ÿ
目录
可以将输入展开成一维的,不对batch size做展开
这个层一般用于将输入特征展开,便于后面接续全连接层。比如,输入数据都是20x20x1的灰度图片,现在想在这个图片后面直接连个全连接层。tf.keras.layers.Flatten就派上用场了,将每张图片展开成400的一维数组。
tf.keras.layers.Flatten不对batch size做展开,比如是如果有(batch_size,20,20)的输入ÿ