Have a question? Connect with the community at the TensorFlow Forum Visit Forum

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.