tf.keras.backend.batch_flatten

View source on GitHub

Turn a nD tensor into a 2D tensor with same 0th dimension.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.keras.backend.batch_flatten

tf.keras.backend.batch_flatten(
    x
)

In other words, it flattens each data samples of a batch.

Arguments
x	A tensor or variable.

Returns
A tensor.

Examples:

Flattening a 3D tensor to 2D by collapsing the last dimension.

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)