Rewrite parser function to enable vectorization

Question

I am working with TensorFlow datasets. For loading image data from a record file, I have a function parse_tfr_elem, which currently only loads one example.