Create a PyTorch data loader

This tutorial can be accessed from the Jupyter notebook: