0
これは非常に簡単かもしれませんが、私は答えを見つけることができません。私はtf.while_loopの 'body'の内側にテンソルを使用しようとしています。簡単にするために、(3,4)型テンソル 'x'をそれに渡しています。しばらくの間、 'body'関数の中で何もしません。しかし、この議論の通過はいくつかの問題を引き起こしているようです。スタックトレースは 'AssertionError:'だけを伝えます。助けてください。 コード:TensorflowのAssertionError while_loop
import tensorflow as tf
import numpy as np
def cond(sequence_len, step, x):
return tf.less(step,sequence_len)
def body(sequence_len, step, x):
return (sequence_len, step+1)
step = tf.constant(0)
sequence_len = tf.constant(10)
x = tf.zeros([3, 4], tf.int32)
res,step = tf.while_loop(cond,body,[sequence_len, step, x])
with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
step_eval = step.eval(session=sess)
print(step_eval)
完全なスタックトレースも下に貼り付けられます。 The image of the stack trace