Define torch dataloader with h5py dataset

[ptrblck]

Have a look at this topic where we debugged some multiprocessing issues with HDF5.