tf.layers.flatten(
inputs,
name=None,
data_format='channels_last'
)
Defined in tensorflow/python/layers/core.py
.
Flattens an input tensor while preserving the batch axis (axis 0). (deprecated)
Arguments:
inputs
: Tensor input.name
: The name of the layer (string).data_format
: A string, one ofchannels_last
(default) orchannels_first
. The ordering of the dimensions in the inputs.channels_last
corresponds to inputs with shape(batch, height, width, channels)
whilechannels_first
corresponds to inputs with shape(batch, channels, height, width)
.
Returns:
Reshaped tensor.
Examples:
x = tf.placeholder(shape=(None, 4, 4), dtype='float32')
y = flatten(x)
# now `y` has shape `(None, 16)`
x = tf.placeholder(shape=(None, 3, None), dtype='float32')
y = flatten(x)
# now `y` has shape `(None, None)`