PyTorch dataset extended with map, cache etc. (tensorflow.data like) - View it on GitHub