class_docstr: Structure which holds the logits and any aux head logits.
class_docstr_tok: ["Structure","which","holds","the","logits","and","any","aux","head","logits","."]
code: def __new__(
      cls,
      logits,
      logits_weight=1.0,
      aux_logits=None,
      aux_logits_weight=None,
  ):
      if aux_logits is not None:
          msg = "aux_logits_weight cannot be None if aux_logits is not None"
          assert aux_logits_weight, msg
      return super(LogitsSpec, cls).__new__(
          cls, logits, logits_weight, aux_logits, aux_logits_weight
      )
  
imports: import collections
  import os
  import string
  import time
  from absl import flags
  from absl import logging
  from model_search import block
  from model_search import block_builder
  from model_search import hparam
  from model_search import utils
  from model_search.proto import hparam_pb2
  from model_search.proto import phoenix_spec_pb2
  from model_search.proto import transfer_learning_spec_pb2
  import numpy
  import tensorflow.compat.v2
  import tf_slim
  from google.protobuf import text_format
label_desc: Use image node_id 1 for calling the LogitsSpec obj's underlying member method code with example usage: obj.__new__(cls, logits, logits_weight, aux_logits, aux_logits_weight) and returns: super