code: def get_predicted_class(self, x, **kwargs):
      
      return tf.argmax(self.get_logits(x, **kwargs), axis=1)
docs: :param x: A symbolic representation (Tensor) of the network input
      :return: A symbolic representation (Tensor) of the predicted label