2017-07-15 143 views

回答

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) 

如果您的起始向量不包括整数,您需要更改的类型。

相关问题