2
甲张量阵列是: 数组= [1,1,0,1,1,0]如何找到Tensorflow最大值的索引,但该值是重复
如果使用tf.argmax(),它只能找到第一个索引。 输出=>“0”
但我想找到在最后指数的最大值。 输出将是“4”
甲张量阵列是: 数组= [1,1,0,1,1,0]如何找到Tensorflow最大值的索引,但该值是重复
如果使用tf.argmax(),它只能找到第一个索引。 输出=>“0”
但我想找到在最后指数的最大值。 输出将是“4”
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)
如果您的起始向量不包括整数,您需要更改的类型。