python-3.x - Tensorflow:使用序列设置数组元素

Question

我正在尝试使用自己的图像数据集训练 CNN，但是当将批处理数据和标签传递到 feed_dict 时，我收到错误 ValueError: seting an array element with a我读到的序列 here ，这是一个维度问题，可能来 self 的 batch_label 张量，但我无法弄清楚如何使其成为一个单热张量(我的图表所期望的)。

我在这里上传了完整的代码作为要点:https://gist.github.com/guivn/f7f753547f77a3b12992

Answer 1

TL;DR:您无法提供 tf.Tensor 对象(即 batch_data 和 batch_labels) > 在 gist 中)作为另一个张量的值。 (我相信在最新版本的 TensorFlow 中，错误消息应该更清楚。)

遗憾的是，您目前无法使用 feed/tf.placeholder() 机制将一个 TensorFlow 图的结果传递给另一个图。我们正在研究如何使这变得更容易，因为这是一个常见的混淆和功能请求。然而，对于您的确切程序来说，解决这个问题应该很容易。只需移动 lines创建输入并用它们替换占位符。您的程序将如下所示:

with graph.as_default():

  # Input data.
  filename_and_label_tensor = tf.train.string_input_producer(['train.txt'], shuffle=True)
  data, label = parse_csv(filename_and_label_tensor)
  tf_train_dataset, tf_train_labels = tf.train.batch([data, label], batch_size, num_threads=4)

  # Rest of the model construction goes here....

通常，如果您想通过同一模型传递另一个数据集，例如用于评估 - 最简单的方法是制作图表的另一个副本(可能共享相同的 tf.Variable 对象)。