How to reshape BatchDataset class tensor?

Question

Im unable to reshape tensor loaded from my own custom dataset. As shown below ds_train has batch size of 8 and I want to reshape it such as: len(ds_train)