Существует ли механизм извлечения предсказаний Кераса как генератора?
В моем случае входные данные поступают через tf.data.Dataset.Если это уместно, код выполняется на TPU (так как это влияет на некоторую поддержку API).Размер массива прогнозирования слишком велик, чтобы поместиться в памяти, поэтому я хотел бы получить к нему доступ в качестве итератора (например, как это можно сделать с помощью TPUEstimator).На данный момент единственный подход, который я могу придумать, состоит в том, чтобы разбить мой входной набор данных на отдельные порции и обрабатывать их отдельно, но это значительно замедляет конвейер ввода данных и кажется ненужным.