tensorflow随笔-条件循环控制（12）

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/u010255642/article/details/82526828

#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
求1+...+5，计算累加
@author: [email protected]
"""
import tensorflow as tf

i = tf.constant(0)
num_sum=tf.Variable(0)

def cond(i,num_sum):
    return tf.less(i,6)

def body(i,num_sum):
    num_sum=tf.add(num_sum,i)
    i=tf.add(i,1)
    return (i,num_sum)


i,num_sum= tf.while_loop(cond, body, [i,num_sum])
with tf.Session() as sess:
    tf.global_variables_initializer().run()
    res = sess.run([i,num_sum])
    print res
    print res[1]
    print 1+2+3+4+5

利用while_loop计算累加结果