テンソルフローのスカラー条件に基づいて選択するにはどうすればよいですか？

Question

条件がスカラーであるが、tとeが等しい形状のテンソルである場合、どうすればtf.select(condition, t, e)ですか？ TensorFlowは、条件がtとeと同じ形状であることを求めています。

Answer 1

旧tf.select() OPは今tf.where()（TensorFlow 1.0）と呼ばれているなどの条件を同じ形状を作るためにtf.tileを使用することができます。 tf.where() opの形状は、conditionの形状がやや奇妙で、2つのブランチとは異なる形状を持つことができますが、その長さが0番目のディメンションと同じサイズのベクトルの場合にのみtとeです。したがって、次のようにプログラムを動作させることができます：

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

あなたはt