Pytorch dataset. Pytorch.