how to store 4d matrixes in tfrecords or how to load them in specific batches

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