Согласно принятому ответу на этот вопрос , экземпляры Tensorflow tf.data.Dataset
оцениваются лениво , что означает, что вам, в принципе, может потребоваться перебрать весь набор данных, чтобы устанавливает sh количество различных меток и форму (ы) входных данных (которые могут быть переменными, например, с последовательностями звука или текста переменной длины).