TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.layers.Flatten

View source on GitHub

Class Flatten

Flattens an input tensor while preserving the batch axis (axis 0).

Inherits From: Flatten, Layer

Aliases:

  • Class tf.compat.v1.layers.Flatten

Arguments:

  • data_format: A string, one of channels_last (default) or channels_first. The ordering of the dimensions in the inputs. channels_last corresponds to inputs with shape (batch, ..., channels) while channels_first corresponds to inputs with shape (batch, channels, ...).

Examples:

  x = tf.compat.v1.placeholder(shape=(None, 4, 4), dtype='float32')
  y = Flatten()(x)
  # now `y` has shape `(None, 16)`

  x = tf.compat.v1.placeholder(shape=(None, 3, None), dtype='float32')
  y = Flatten()(x)
  # now `y` has shape `(None, None)`

__init__

View source

__init__(
    data_format=None,
    **kwargs
)

Properties

graph

DEPRECATED FUNCTION

scope_name