python - 类型错误 : 'Tensor' object cannot be interpreted as an integer

Question

我正在尝试根据数组的大小运行一个循环。如何在 tensorflow 中做到这一点？例如

# input pipeline with all files available in the folder
a = tf.Variable([1,2,3,4,5],dtype = tf.int32)
loop = tf.size(a)
....
for i in range(loop):
    print(sess.run(a))

我想打印数组 a 5 次。但它说循环是一个张量对象，不能作为整数。我尝试将循环变量作为

loop = tf.cast(tf.size(a),tf.int32),
loop = tf.shape_n(a),
loop = tf.shape(a)[0]

同样的错误。

Answer 1

不太确定你想在这里实现什么。 loop 是一个 tf.Tensor 而 range 需要一个 integer 作为参数，因此会出现错误。如果只想打印 5 次 a，为什么不直接将 loop 设置为数值 5？

否则，下面的代码应该可以工作，因为 loop.eval() 返回 loop 的值 5:

a = tf.Variable([1,2,3,4,5],dtype = tf.int32)
loop = tf.size(a)
....
for i in range(loop.eval()):
    print(sess.run(a))

如果不想多次执行TF图，看看tf.while_loop .