Как сохранить объект tf.data.Dataset?

Question

Как указано выше. Я попытался засолить это, но я получил эту ошибку:

Maybe_arr = self._cpu_nograd()._numpy() # pylint: disable= защищенный доступ

RuntimeError: тип Tensorflow 21 не может быть преобразован в numpy dtype.

Answer 1

tf.data.Dataset- это более абстрактный объект, задача которого - определить конвейер данных. Если вы хотите сохранить промежуточные результаты, чтобы ускорить конвейер данных, вы можете использоватьtf.data.Dataset.cache() или tf.data.Dataset.prefetch() (подробнее об этом здесь)

Если вы заинтересованы в сохранении последовательности операций в конвейере данных, я предполагаю, что такого нет, и вам нужно сохранить код для конвейера данных. Я не знаю ни одного метода, который мог бы извлечь график конвейера данных Dataset API. Если кому-то известно об этом, добавьте к ответу.