- Initialize one partitioner for each batch. - Collect partition size during initialization. - Support base ridx in the finalization.
- Initialize one partitioner for each batch. - Collect partition size during initialization. - Support base ridx in the finalization.