如何找到Tensorflow最大值的索引，但該值是重複

Question

甲張量陣列是： 數組= [1，1，0，1，1，0]

如果使用tf.argmax（），它只能找到第一個索引。 輸出=>「0」

但我想找到在最後指數的最大值。 輸出將是「4」

Answer 1

tf.argmax不會返回第一個最大值。如果領帶任何東西都可以退回：

請注意，在綁定的情況下返回值的身份不是 保證。

所以像reverse和argmax這樣的答案是錯誤的。

一種選擇，我可以看到的是：

import tensorflow as tf 
a = tf.constant([5, 3, 3, 5, 4, 2, 5, 1]) 
b = tf.argmax(tf.multiply(
    tf.cast(tf.equal(a, tf.reduce_max(a)), tf.int32), 
    tf.range(1, a.get_shape()[0] + 1) 
)) 
with tf.Session() as sess: 
    print sess.run(b) 

如果您的起始向量不包括整數，您需要更改的類型。