storing 4d arrays as tfrec or retrieve them as a batch

Question

ok so I am not very used to working with TensorFlow but i am playing around with TPUS and ran into an annoying problem.

I have the following:

A 4D array (an image