Function compute_accuracy

tensorflow/examples/tf2_showcase/mnist.py:131–135

(logits, labels)

def compute_accuracy(logits, labels):
  predictions = tf.argmax(logits, axis=1, output_type=tf.int64)
  labels = tf.cast(labels, tf.int64)
  return tf.reduce_mean(
      tf.cast(tf.equal(predictions, labels), dtype=tf.float32))


# TODO(brianklee): Enable @tf.function on the training loop when zip, enumerate