ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: oryx.experimental.nn.reshape

Contains layers that reshape arrays.

Classes

class Flatten: Flattens the inputs collapsing all ending dimensions.

class Reshape: Reshape the inputs to a new compatatible shape.