如何根据tensorflow中的标量条件进行选择？

Question

如果条件是标量，但t和e是相等形状的张量，那么我该怎么做tf.select(condition, t, e)？ TensorFlow希望条件与t和e相同。

Answer 1

前者tf.select()运现在被称为tf.where()（在TensorFlow 1.0）。 tf.where() op在condition的形状上有一个稍微奇怪的情况：它可以具有与两个分支不同的形状，但只有它的长度与t和e的第0维尺寸相同的向量才具有不同的形状。因此，你可以使你的程序工作如下：

condition = ... # scalar 
t = ...   # shape = [4, 9, 2] 
e = ...   # shape = [4, 9, 2] 

# Tile `condition` as a vector whose length matches the 0th dimension of `t`. 
condition_vector = tf.tile([condition], [tf.shape(t)[0]]) 

result = tf.where(condition_vector, t, e) 

Answer 2

您可以使用tf.tile使条件相同的形状t