tf.while_loop - ValueError：这两个结构的元素数量不同

Posted 2021-05-04

我正在尝试在Tensorflow中执行以下操作 -

import tensorflow as tf

graph = tf.Graph()

with graph.as_default():

    i = tf.Variable(0)
    sol = tf.Variable(0)

    def cond(i, sol):
        return tf.less(i, 2)
    def body(i, sol):
        i = tf.add(i, 1)
        sol = tf.add(sol, 1)
    tf.while_loop(cond, body, [i, sol])

with tf.Session(graph=graph) as session:
    tf.global_variables_initializer().run()

    result = session.run(sol, feed_dict={})
    print result

我得到的错误如下图所示。

我无法理解错误消息中的两个“结构”是什么。我想最终根据tf.Placeholder（上面代码中的'i'）的值创建一个'tf.while_loop'和'condition'。

答案

你应该将return添加到body函数：

def body(i, sol):
    i = tf.add(i, 1)
    sol = tf.add(sol, 1)
    retrun [i, sol]

但我认为您还应该将代码更改为类似的代码

graph = tf.Graph()

with graph.as_default():
    i = tf.Variable(0)
    sol = tf.Variable(0)

    def cond(i, sol):
        return tf.less(i, 2)
    def body(i, sol):
        i = tf.add(i, 1)
        sol = tf.add(sol, 1)
        return [i, sol]
    result = tf.while_loop(cond, body, [i, sol])

with tf.Session(graph=graph) as session:
    tf.global_variables_initializer().run()

    result = session.run(result, feed_dict={})
    print(result[1])

因为tf.while只是图表中的节点，你应该运行它，否则你将得不到任何结果。