class_docstr: Reversible PrintShape for debugging reversible serial layers.
class_docstr_tok: ["Reversible","PrintShape","for","debugging","reversible","serial","layers","."]
code: def __init__(self, n_in=1, msg=""):
      super().__init__(n_in=n_in, n_out=n_in)
      self._msg = msg
  
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 ReversiblePrintShape object from inherited base classes: ReversibleLayer with example: obj = ReversiblePrintShape(n_in, msg)