How can I transform my Tensorflow dataset to another shape?

Question

I have created a dataset with make_csv_dataset which has the specs as under

print(trains_ds.element_spec)

(OrderedDict([(\'gaze_