如何從張量張量中得到子張量？

Question

label = tf.constant([0,1,2,3,4,4,5,5]) 

我有一個張量，例如，一個以上。 我想過濾元素爲4的張量。輸出張量應該是[4,4]。 如何實現它？謝謝。

Answer 1

只需使用tf.where獲得其中一個條件是真實的指數，並tf.gather收集指定的值

import tensorflow as tf 

label = tf.constant([0,1,2,3,4,4,5,5]) 
filtered = tf.gather(label, tf.where(tf.equal(label, 4))) 
sess = tf.Session() 
print(sess.run(filtered)) 

[4] [4]