Save the date! Google I/O returns May 18-20 Register now

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.