私はTensorArrayとWHILE_LOOPの組み合わせのために非常に簡単な例を生成しようとしています:TensorArrayとwhile_loopはテンソルフローでどのように連携しますか?
# 1000 sequence in the length of 100
matrix = tf.placeholder(tf.int32, shape=(100, 1000), name="input_matrix")
matrix_rows = tf.shape(matrix)[0]
ta = tf.TensorArray(tf.float32, size=matrix_rows)
ta = ta.unstack(matrix)
init_state = (0, ta)
condition = lambda i, _: i < n
body = lambda i, ta: (i + 1, ta.write(i,ta.read(i)*2))
# run the graph
with tf.Session() as sess:
(n, ta_final) = sess.run(tf.while_loop(condition, body, init_state),feed_dict={matrix: tf.ones(tf.float32, shape=(100,1000))})
print (ta_final.stack())
をしかし、私は次のエラーを取得しています:
ValueError: Tensor("while/LoopCond:0", shape=(), dtype=bool) must be from the same graph as Tensor("Merge:0", shape=(), dtype=float32).
誰もが問題が何であるかという考えに持っていますか?
:ここ
は、実施例です'session.run(TensorArray)'。 – sirfz
申し訳ありませんが、私はあなたが意味するものを得ていませんでした。あなたは正しい書式を書いてくださいますか? –