python - 如何选择张量中的一组项目

Question

在 tensorflow 中。如何选择所有三元组 (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)