Module: oryx.experimental.nn.reshape

Stay organized with collections Save and categorize content based on your preferences.

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.