1. Run the softmax activation function on the output of the second fully connected layer:

y_pred = tf$nn$softmax(layer_fc2_drop)

2. Use the argmax function to determine the class number of the label. It is the index of the class with the largest (probability) value:

y_pred_cls = tf$argmax(y_pred, dimension=1L)