如何在张量流中累积1到10?

时间:2018-10-17 11:53:21

标签: python tensorflow

我想在张量流中累积1到10。但这行不通。 谁能帮我解决这个问题。

one = tf.constant(1)
value = tf.Variable(0,name="value")
increasing_op = tf.add(value,one)
assign_op = tf.assign(value,increasing_op)

#sum
sumvalue = tf.Variable(0,name = "sumvalue")
sum_op = tf.add(sumvalue,assign_op)

init = tf.global_variables_initializer()

with tf.Session() as session:
    session.run(init)
    for _ in range(10):
        session.run(sum_op)
    print(session.run(sumvalue))

2 个答案:

答案 0 :(得分:1)

sum_op返回计算后的值。您也可以利用tf.assign_add

sumvalue = tf.Variable(0,name = "sumvalue")
sum_op = tf.assign_add(sumvalue, 1)

init = tf.global_variables_initializer()

with tf.Session() as session:
    session.run(init)
    for _ in range(10):
        sum_value = session.run(sum_op)
    print(sum_value)

答案 1 :(得分:0)

感谢您的帮助。 我修改了如下代码

model

是否可以将占位符用于n