如何使用數據集管道推斷張量流中的單個示例

Question

我用可變長度的輸入向量訓練dnn。 因此，我使用TFRecord文件和feature_lists創建了數據集管道。 可以了 通常我使用feed_dict {...}來獲取單個示例的“答案”。 使用變量輸入現在很難實現。 所以我想知道是否有一個很好的方法可以在不使用單個示例的情況下提供數據集管道：

dataset = tf.data.TFRecordDataset(data_path)

但是相反，我可以在其中放一個示例，然后使用我的常規（且正常工作）的數據集管道處理預測。

一種解決方法是將單個示例保存到TFRecord文件，然后從文件中讀取此數據。 但是我認為有另一種方法可以做到這一點？

謝謝 ：）

Answer 1

您可以使用tf.placeholder_with_default ，並且默認情況下，您傳遞iterator.get_next()結果。 有關更多信息，請參見這篇文章： 將iterator.get_next（）分配為默認的占位符，並使用默認值