О формате входных данных тензора. Это делает PNG-изображение напрямую форматом TFrecord. Или сделать png-образный формат npy, а затем заставить npy стать TFrecord?
Как вы читаете все примеры из TFRecords одновременно? Я использовал tf.parse_single_example для считывания отдельных примеров, используя код, аналогичный приведенному в методе read_and_decode в ...