I am expecting to get a batch for each sequence length
for example one batch could be of shape : [16 , 80 , 2048] another batch would be of also [16 , 80 , 2048] or [16,120,2048] I thought about sorting the samples by lengths but I want to shuffle them!