code: def _collect_layer_outputs(mod, data, include_layer=None, max_num_examples=None, logger=None):
      
      collector = _LayerOutputCollector(include_layer=include_layer, logger=logger)
      num_examples = _collect_layer_statistics(mod, data, collector, max_num_examples, logger)
      return collector.nd_dict, num_examples
docs: Collect layer outputs and save them in a dictionary mapped by layer names.