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.
x: A tensor or variable.A tensor.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/backend/batch_flatten