Como você lê todos os exemplos de um TFRecords de uma só vez?

Eu tenho usadotf.parse_single_example para ler exemplos individuais usando código semelhante ao fornecido no métodoread_and_decode noexemplo do full_connected_reader. No entanto, quero executar a rede em todo o conjunto de dados de validação de uma só vez e, portanto, gostaria de carregá-los na íntegra.

Não tenho muita certeza, masa documentação parece sugerir que eu posso usartf.parse_example ao invés detf.parse_single_example para carregar o arquivo TFRecords inteiro de uma só vez. Parece que não consigo fazer isso funcionar. Suponho que isso tenha a ver com a forma como especifico os recursos, mas não sei como, na especificação do recurso, declarar que existem vários exemplos.

Em outras palavras, minha tentativa de usar algo semelhante a:

reader = tf.TFRecordReader()
_, serialized_example = reader.read(filename_queue)
features = tf.parse_example(serialized_example, features={
    'image_raw': tf.FixedLenFeature([], tf.string),
    'label': tf.FixedLenFeature([], tf.int64),
})

não está funcionando e suponho que é porque os recursos não estão esperando vários exemplos ao mesmo tempo (mas, novamente, não tenho certeza). [Isso resulta em um erro deValueError: Shape () must have rank 1]

Essa é a maneira correta de ler todos os registros de uma só vez? E se sim, o que preciso alterar para realmente ler os registros? Muito obrigado!