如何选择张量中的项目组

Question

在张量流中。 如何选择所有三元组，（x, y, c）c > 0.5

我知道这可能是一个非常基本的问题，但我对Tensorflow来说是一个新手。

Answer 1

使用tf.where。例如，

x = np.random.rand(20,3)
sess = tf.Session()
print x[tf.where(tf.greater(x[:,2], 0.5)).eval(session=sess)]

或略微清洁， tf.boolean_mask(x,tf.greater(x[:,2], 0.5)).eval(session=sess)