megatron.data.dataset_utils.get_a_and_b_segments#

megatron.data.dataset_utils.get_a_and_b_segments(sample, np_rng)#

Divide sample into a and b segments.