tf.keras.layers.Flatten
| Просмотреть исходный код на GitHub |
Распаковывает входные данные. Не влияет на размер пакета.
Наследуется от: Layer
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 = Sequential()
model.add(Convolution2D(64, 3, 3,
border_mode='same',
input_shape=(3, 32, 32)))
# now: model.output_shape == (None, 64, 32, 32)
model.add(Flatten())
# now: model.output_shape == (None, 65536)
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/layers/Flatten