tf.keras.backend.batch_flatten

Преобразует тензор nD в тензор 2D с тем же нулевым измерением.

tf.keras.backend.batch_flatten(
    x
)

Другими словами, она сглаживает каждый образец данных в пакете.

Примеры:

Сглаживание 3D тензора до 2D, сворачивая последнее измерение.

x_batch = tf.keras.backend.ones(shape=(2, 3, 4, 5))
x_batch_flatten = batch_flatten(x_batch)
tf.keras.backend.int_shape(x_batch_flatten)
(2, 60)