https://travelebookdownload.blogspot.com/2023/12/how-to-convert-tensorflow-dataset-to-2d.html