tensorflow for循环 while循环 实例

tensorflow for循环 while循环 实例

import tensorflow as tfn1 = tf.constant(2)n2 = tf.constant(3)n3 = tf.constant(4)def cond1(i, a, b): return i < n1def cond2(i, a, b): return i < n2def cond3(i, a, b): return i < n3def body(i, a, b): return i + 1, b, a + bi1, a1, b1 = tf.while_loop(cond1, body, (2, 1, 1))i2, a2, b2 = tf.while_loop(cond2, body, (2, 1, 1))i3, a3, b3 = tf.while_loop(cond3, body, (2, 1, 1))sess = tf.Session()print(sess.run(i1))print(sess.run(a1))print(sess.run(b1))print("-")print(sess.run(i2))print(sess.run(a2))print(sess.run(b2))print("-")print(sess.run(i3))print(sess.run(a3))print(sess.run(b3))

print结果：

211-312-423

可见body函数返回的三个变量又传给了body

改自 ​​https://github.com/vahidk/EffectiveTensorflow​​

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。