python - 如何使用提供给 tf.placeholder(tf.int32) 的整数？

Question

我需要初始化一个动态列表，其长度由一个整数给出，我将通过 feed_dict 将其放入占位符中。

我通过以下代码设置了图表:

num_placeholder = tf.placeholder(tf.int32) 
input_data = list() 
for _ in range(num_placeholder):
    input_data.append(1) 

但出现以下错误:“Tensor”对象无法解释为整数

如何使用放入 num_placeholder 中的 int？

Answer 1

num_placeholder是 Tensor对象，它不能直接用作int ，必须首先对其进行评估。请参阅有关 placeholder 的文档方法。

如果您想使用num_placeholder对象为int您必须按如下方式对其进行评估:

# create a placeholder       
num_placeholder = tf.placeholder(tf.int32)

# create an identity operation
# because same Tensor can not be both fed and fetched
num_op = tf.identity(num_placeholder)

with tf.Session() as sess:
    # feed value to placeholder and fetch it
    num = sess.run([num_op], feed_dict={num_placeholder: 5})

现在num will 是一个列表 [5]你可以使用 num[0]在你的循环中。我不太确定您想要实现什么，但如果您的目标是创建一个在运行时馈送的某些自定义长度的向量，则以下是如何通过 Tensorflow 解决它:

num_placeholder = tf.placeholder(tf.int32)
ones_op = tf.ones(shape=[num_placeholder], dtype=tf.int32)
with tf.Session() as sess:
    ones = sess.run([ones_op], feed_dict={num_placeholder: 5})

ones ==> <type 'list'>: [array([1, 1, 1, 1, 1], dtype=int32)] ,请参阅 ones