https://capressocoffemakerthecheapest.blogspot.com/2024/05/tensorflow-datasets-reshape-images.html