python - 为什么 TensorFlow Estimator API 将输入作为 lambda？

Question

tf.estimator API 采用返回 Dataset 的输入“输入函数”。例如，Estimator.train() 采用 input_fn ( documentation )。

在我见过的示例中，无论何时手动提供此函数，它都是无参数的 lambda。

那不是说函数总是返回相同的值吗？还是在没有参数的情况下多次调用？我找不到关于此的文档。为什么像 train() 这样的函数不直接将输入作为 Dataset 显式接收？

Answer 1

Dataset 对象也由计算图中的节点支持。 Estimator 为每个 train()、evaluate() 等构建计算图。通过这样做，Estimator API 确保对不同 Estimator 对象的操作是隔离的，并将 Tensor/Dataset 放入独立的图形中。

一些代码指针(针对 TensorFlow 1.7)以供您感兴趣:

• Estimator.train() 最终调用 this
• 依次invokes input_fn

希望对您有所帮助。