DataLoader can’t take tuples of tensors when the tensors are in different shapes

Collecting samples with the same or similar size might be a good idea and @vdw shared some ideas and experiments here.