标签： linear-algebra tensorflow
如何在TensorFlow中获取沿选定轴的张量最大元素的索引？
答案 0 :(得分：13)
对不起，已经想通了。它是：
tf.argmax(input, axis, name=None)
https://www.tensorflow.org/api_docs/python/tf/argmax