テンソルのリストをテンソルフローのグラフの入力として渡すにはどうすればよいですか？

Question

私はtf.placeholder()の形状[None, None, 10]を使ってテンソルに入力を渡します。今、私は入力の最初の次元を反復し、その次元の各スライスにある関数を適用したいと思います。しかし、私がPython forループを使用してこれを実行しようとすると、Tensorオブジェクトが "iterable"ではないというエラーが表示されます。

テンソルのリストとして入力を渡す方法はありますか[None, 10]、このリストをプレースホルダにどのように割り当てることができますか？または、Tensorのディメンションを繰り返し処理する別の方法がありますか？

Answer 1

テンソルxの形を[None、None、10]の形で渡してから、tf.split(0, -1, x)を使用して反復処理できるテンソルのリストを取得する必要があります。

Answer 2

これは新しいtf.map_fn()、tf.foldl()tf.foldr()又は（最も一般的）バージョン0.8にTensorFlowに加えtf.scan()高次演算子を使用して可能です。使用する特定の演算子は、実行する計算によって異なります。たとえば、あなたがテンソルの各列に同じ機能を実行し、バック単一テンソルに要素をパックしたい場合、あなたはtf.map_fn()を使用します。

p = tf.placeholder(tf.float32, shape=[None, None, 100]) 

def f(x): 
    # x will be a tensor of shape [None, 100]. 
    return tf.reduce_sum(x) 

# Compute the sum of each [None, 100]-sized row of `p`. 
# N.B. You can do this directly using tf.reduce_sum(), but this is intended as 
# a simple example. 
result = tf.map_fn(f, p)