def __init__(self, name, n_in, num_summaries=5,
               recover_fn=None):
    """Takes a tensor and returns it.

    Args:
      name: Name of the metric to be reported.
      n_in: Number of inputs.
      num_summaries: Number of images to show.
      recover_fn: the function for converting a tensor to a dipslayable image.
    """
    super().__init__(name=f'Summary_{name}', n_in=n_in, n_out=n_in)
    name = 'summary_' + name
    self._name = name
    self._num_summaries = num_summaries
    self._recover_fn = recover_fn
Takes a tensor and returns it.

    Args:
      name: Name of the metric to be reported.
      n_in: Number of inputs.
      num_summaries: Number of images to show.
      recover_fn: the function for converting a tensor to a dipslayable image.
    
trax/layers/core.py