如何使用循环在tensorflow中存储操作？

import tensorflow as tf

# 定义循环条件函数
def condition(i, x):
    return tf.less(i, 10)

# 定义循环体函数
def body(i, x):
    # 执行操作
    x = tf.add(x, i)
    # 更新迭代变量
    i = tf.add(i, 1)
    return i, x

# 初始化迭代变量和存储变量
i = tf.constant(0)
x = tf.constant(0)

# 执行循环
i, x = tf.while_loop(condition, body, [i, x])

# 创建会话并运行计算图
with tf.Session() as sess:
    result = sess.run(x)
    print(result)