Цикл по tf.data.Dataset очень медленно - PullRequest
1 голос
/ 18 июня 2020

Мне интересно, почему for-l oop по выборкам tf.data.Dataset намного медленнее, чем цикл по соответствующему массиву numpy.

import numpy as np
import tensorflow as tf
import time

a = np.ones(100000, dtype=np.float32)

start_time = time.time()
for x in a:
    pass
print(time.time() - start_time)

start_time = time.time()
for x in tf.data.Dataset.from_tensor_slices(a):
    pass
print(time.time() - start_time)

0.05548405647277832
5.67711615562439

Моя версия TensorFlow - 2.0.0.

...