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.