`torch.chunk` source code

I can’t seem to find the source code for torch.chunk in PyTorch’s Github page or in the documentation. Anyone knows where this is in PyTorch’s Github page?

Many thanks in advance.


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

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

Oh I see. Many thanks for this!