Thanks buddy. I was asking for the source code in Github, not documentation.
Anyhow, I got it here now: https://github.com/pytorch/pytorch/blob/master/torch/csrc/api/include/torch/data/datasets/chunk.h 8