tf.keras.backend.batch_flatten(x)
Defined in tensorflow/python/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.