TensorFlow张量中最大元素的索引

Ser*_* S. 5 linear-algebra tensorflow

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

Ser*_* S. 13

对不起,已经想通了.它是:

tf.argmax(input, axis, name=None)
Run Code Online (Sandbox Code Playgroud)

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