https://travelebookdownload.blogspot.com/2023/04/how-to-customize-pytorch-data.html