fixed dataset to no longer incorrectly shuffle the batch dimensions