tensorflow，如何索引張量的真實值？

Question

我有一個一維張量，像：

[false, false, true, false, true, false]

如何找到所有true值的索引？

我的解決方案是：

1. 將其設為1和0值
2. 使用argmax API查找一個索引，然后將其設置為false / 0 。
3. 再次使用argmax查找下一個true / 1值

但是這種解決方案不是很好。

Answer 1

import tensorflow as tf

a = tf.constant([False,False,True,False,True],dtype=tf.bool)
b = tf.where(a)
sess = tf.Session()
print(sess.run(b))

Answer 2

這是您要找的東西嗎？ [k for k, value in enumerate(tensor) if value]

Answer 3

In [1]: import tensorflow as tf                                                                                                                                                                                      

In [2]: a = tf.constant([False, False, True, True])

In [3]: a_n = [tf.cond(tf.equal(v, tf.constant(True)), lambda: tf.constant(k), lambda: tf.constant(-1)) for k, v in enumerate(tf.unstack(a))]                                                                        

In [4]: sess = tf.Session()

In [5]: sess.run(a_n)                                                                                                                                                                                                
Out[5]: [-1, -1, 2, 3]

希望這可以幫助...