Index of a maximum element in TensorFlow tensor

Question

How can I get index of a maximum element of a tensor along a selected axis in TensorFlow?

Answer 1

Sorry, already figured that out. It is:

tf.argmax(input, axis, name=None)

https://www.tensorflow.org/api_docs/python/tf/argmax