make collator flexible, repeat same dims for random