TensorFlow 张量中最大元素的索引

Index of a maximum element in TensorFlow tensor

如何在 TensorFlow 中获取张量沿选定轴的最大元素的索引？

抱歉，我已经知道了。它是：

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

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