tf.keras.layers.Reshape

Слой, преобразующий входные данные в заданную форму.

Наследуется от: Layer

tf.keras.layers.Reshape(
    target_shape, **kwargs
)

Форма входных данных:

Произвольная, хотя все размерности в форме входных данных должны быть известны/фиксированы. Используйте ключевое слово input_shape (кортеж целых чисел, не включая ось образцов/пакета), когда этот слой используется в качестве первого слоя в модели.

Форма выходных данных:

(batch_size,) + target_shape

Пример:

# as first layer in a Sequential model
model = tf.keras.Sequential()
model.add(tf.keras.layers.Reshape((3, 4), input_shape=(12,)))
# model.output_shape == (None, 3, 4), `None` is the batch size.
model.output_shape
(None, 3, 4)

# as intermediate layer in a Sequential model
model.add(tf.keras.layers.Reshape((6, 2)))
model.output_shape
(None, 6, 2)

# also supports shape inference using `-1` as dimension
model.add(tf.keras.layers.Reshape((-1, 2, 2)))
model.output_shape
(None, 3, 2, 2)