tensorflow-通过while_loop计算累加

计算累加

#!/usr/bin/env python2

# -*- coding: utf-8 -*-
"""
Created on Mon Jul 24 08:25:41 2017
求1+...+5
@author: [email protected]
"""
import tensorflow as tf

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

def cond(i,num_sum):
    return i <=5

def body(i,num_sum):
    num_sum+=i
    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

tensorflow的循环

执行结果

[6, 15]
15
15