他のテンソル次元のサイズの範囲

Question

次のような問題でテンソルのサブセットを計算したいと思います。 テンソルTとテンソルインデックスがIです。

C[i] = T[i, I[i]] 

は私がやると思ったことを実行します：私はテンソルようC取得したいと思い

first_dim = I.get_shape()[0] 
R = tf.range(first_dim) 
S = tf.stack([R, I], 1) 
tf.gather(T, S) 

をしかし、私は2行目の問題があります。

ValueError: Cannot convert an unknown Dimension to a Tensor: ? 

を

Iの形状は、一部のプレースホルダーのサイズによって異なります。私の問題を解決したり、TF-graphのコンパイル時に知られていない別のテンソルの次元の範囲を取得する別の方法はありますか？

Answer 1

get_shapeの代わりにtf.shapeを使用してください。