contrib.layers.flatten
tf.contrib.layers.flatten
tf.contrib.layers.flatten
flatten( inputs, outputs_collections=None, scope=None )
Defined in tensorflow/contrib/layers/python/layers/layers.py
.
See the guide: Layers (contrib) > Higher level ops for building neural network layers
Flattens the input while maintaining the batch_size.
Assumes that the first dimension represents the batch.
Args:
-
inputs
: A tensor of size [batch_size, ...]. -
outputs_collections
: Collection to add the outputs. -
scope
: Optional scope for name_scope.
Returns:
A flattened tensor with shape [batch_size, k].
Raises:
-
ValueError
: If inputs rank is unknown or less than 2.
© 2017 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/contrib/layers/flatten