tf.keras.layers.Reshape

Layer that reshapes inputs into the given shape.

Inherits From: Layer, Module

Used in the notebooks

Used in the guide Used in the tutorials

Input shape:

Arbitrary, although all dimensions in the input shape must be known/fixed. Use the keyword argument input_shape (tuple of integers, does not include the samples/batch size axis)