code: def get_logits(self, x):
      
      logits_name = self._get_logits_name()
      logits_layer = self.get_layer(x, logits_name)
  
      # Need to deal with the case where softmax is part of the
      # logits layer
      if logits_name == self._get_softmax_name():
        softmax_logit_layer = self.get_layer(x, logits_name)
  
        # The final op is the softmax. Return its input
        logits_layer = softmax_logit_layer._op.inputs[0]
  
      return logits_layer
docs: :param x: A symbolic representation of the network input.
      :return: A symbolic representation of the logits