class_docstr: Bundles together logits correctly for distillation.
docstr: Initializes a new Distiller object.
  
  Args:
    distillation_spec: The spec defined in the Phoenix spec.
docstr_tok: ["Initializes","a","new","Distiller","object",".","Args",":","distillation_spec",":","The","spec","defined","in","the","Phoenix","spec","."]
code: def __init__(self, distillation_spec):
      """Initializes a new Distiller object.
  
      Args:
        distillation_spec: The spec defined in the Phoenix spec.
      """
      self._distillation_spec = distillation_spec
  
file_path: model_search/model_search/meta/distillation.py
imports: import collections
  import functools
  from model_search import logit_bundler
  from model_search.architecture import architecture_utils
  from model_search.proto import distillation_spec_pb2
  import tensorflow.compat.v2
label_desc: Use image node_id 1 to create a new Distiller object from inherited base classes: logit_bundler with example: obj = Distiller(distillation_spec)