tf.argmax(input, axis=None, name=None, dimension=None) Returns the index with the largest value across axes of a tensor.