tf.keras.backend.batch_flatten

tf.keras.backend.batch_flatten(x)

Defined in tensorflow/python/keras/_impl/keras/backend.py.

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

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

Arguments:

  • x: A tensor or variable.

Returns:

A tensor.