class_docstr: Reversible reshaping layer.
class_docstr_tok: ["Reversible","reshaping","layer","."]
code: def __init__(self, shape1, shape2, n_in=1):
      self._shape1 = list(shape1)
      self._shape2 = list(shape2)
      name = "ReversibleReshape_%s_%s" % (str(shape1), str(shape2))
      super().__init__(n_in=n_in, n_out=n_in, name=name)
  
imports: from absl import logging
  import jax
  from trax import fastmath
  from trax.layers import base
  from trax.layers import combinators
label_desc: Use image node_id 1 to create a new ReversibleReshape object from inherited base classes: ReversibleLayer with example: obj = ReversibleReshape(shape1, shape2, n_in)