https://cerwinvegassalee.blogspot.com/2023/08/tensorflow-keras-with-tf-dataset-input.html