tf.keras.layers.Flatten
| Просмотреть исходный код на GitHub |
Расплющивает входные данные. Не влияет на размерность пакета.
tf.keras.layers.Flatten(
data_format=None, **kwargs
)
Примечание: Если входные данные имеют форму(batch,)без оси признаков, то расплющивание добавляет дополнительную размерность канала, а форма выходных данных равна(batch, 1).
| Аргументы | |
|---|---|
data_format | Строка, одна из channels_last (по умолчанию) или channels_first. Порядок измерений ввода. channels_last соответствует входам с формой (batch, ..., channels), а channels_first соответствует входам с формой (batch, channels, ...). По умолчанию используется значение image_data_format в файле конфигурации Keras по адресу ~/.keras/keras.json. Если значение никогда не устанавливалось, оно будет равно "channels_last". |
Пример:
model = tf.keras.Sequential() model.add(tf.keras.layers.Conv2D(64, 3, 3, input_shape=(3, 32, 32))) model.output_shape (None, 1, 10, 64)
model.add(Flatten()) model.output_shape (None, 640)
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/keras/layers/Flatten