`torch.chunk` source code

1

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.

2

https://pytorch.org/docs/stable/generated/torch.chunk.html

3

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

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

5

Oh I see. Many thanks for this!