tensorflow：如何选择条件矩阵行

Question

我找到了类似的解决方案here。然而，该解决方案基于一个条件在两个矩阵之间进行二进制选择。我需要做的是只选择满足条件的行。我怎样才能做到这一点？

我将简要介绍问题here。例子是：

tf.InteractiveSession()
yt = tf.constant([10,1,10])
a = tf.constant([1,2,3])
b = tf.constant([3,4,5])
tf.where(tf.less(yt,[5]), a, b).eval()

如果关联的a值小于5，则结果将从yt中选择一个值，否则从b中选择一个值。我需要的是从a yt<5中选择一个值，否则不要选择任何内容。

我怎样才能做到这一点？

谢谢！

Answer 1

你可以这样做：

a[tf.squeeze(tf.where(tf.less(yt,[5]), None, None))