Thank you @ptrblck
I am getting the proper tensors for my images tensor [batch,channel,width,height]  and also getting my label tensor but When I am trying to feed that into my model I am getting the following error.
Screenshot (101)1920×908 67.7 KB