:param x: A symbolic representation (Tensor) of the network input
    :return: A symbolic representation (Tensor) of the predicted label

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