如何保存tf.data.Dataset对象？

Question

如上所述。我尝试腌制，但出现此错误：

  

maybe_arr = self._cpu_nograd（）._ numpy（）＃pintint：   disable =受保护的访问

     

RuntimeError：Tensorflow类型21不能转换为numpy dtype。

Answer 1

tf.data.Dataset是一个更抽象的对象，其任务是定义数据管道。如果要保存中间结果以加快数据流水线速度，则可以使用tf.data.Dataset.cache()或tf.data.Dataset.prefetch() (more on it here)

如果您有兴趣在数据管道中保存操作顺序，那么我认为没有这种事情，您需要保留数据管道的代码。我不知道有什么方法可以提取Dataset API的数据管道图。如果有人知道，请添加答案。