class_docstr: Reversible reshaping layer.
class_docstr_tok: ["Reversible","reshaping","layer","."]
code: def forward(self, inputs):
      if not isinstance(inputs, (tuple, list)):
          inputs = (inputs,)
      res = []
      for x in inputs:
          new_shape = self._shape2 + list(x.shape)[len(self._shape1) :]
          res.append(fastmath.numpy.reshape(x, new_shape))
      if len(res) == 1:
          return res[0]
      return tuple(res)
  
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 2 for calling the ReversibleReshape obj's underlying member method code with example usage: obj.forward(inputs) and returns: tuple, res