TensorFlow:使用sess.run评估单个输入

时间:2016-12-27 13:31:37

标签: tensorflow

我使用会话

在TensorFlow中训练了一个网络
sess.run([train_op, loss], feed_dict=feed_dict)

为会话输入单个数据行并打印输出的最简单方法是什么?

我尝试了(很多变种)

sess.run(print_function, data_row)

但我得到了结果

You must feed a value for placeholder tensor 'Placeholder_1'
with dtype int32 and shape [<batch size>]

1 个答案:

答案 0 :(得分:1)

假设 - 您的第一个维度代表输入数量(如大多数官方示例中的情况)

我建议将占位符的第一个维度设为None,以便可以选择传递任意数量的批次。请查看本教程中的示例 - http://learningtensorflow.com/lesson4/

引用本教程

  

占位符的第一个维度是None,这意味着我们可以有任意数量的行。第二个维度固定为3,这意味着每行需要有三列数据。

https://www.tensorflow.org/resources/faq#how_do_i_build_a_graph_that_works_with_variable_batch_sizes中的官方文档中也有详细记录。