tensorflow - Dataset.map 可以以某种方式并行化 tf.py_func 调用吗？

Question

如本文所述，Dataset.map不支持tf.py_funcmap 函数中调用的并行性（该tf.py_func部分将按顺序执行）。

这个问题有什么解决方法吗？通常情况下，输入管道需要复杂的 python 控制流和 tensorflow 中不存在的操作。过去我曾经使用 python 库实现自己的队列和预处理线程，但是如果能够使用方便的接口来实现它会非常方便tf.data。

标签： tensorflow