Solved by ebarsoum in post #4 
              
              
                The above link is for the Chunk data reader and not torch.chunk. Torch.chunk code is in: https://github.com/pytorch/pytorch/blob/master/aten/src/ATen/native/TensorShape.cpp#L496