PyTorch provides an implementation of parallel_for at Parallel.h. In this signature we do not have any provision for Partioner however, in the Microsoft Docs there is a concept of _Partitioner.
Is there a way we can partion the range into uneven sub-ranges for the threads to work on? In LibTorch?