There won’t be any negative values. If you are saying to use torch.split() then I feel that we will have to pass the list of size of every chunk. Is there any way to calculate the size of chunk other than naive for loop?