There is currently an issue with a missmatch of shapes during batch creation. Therefore add more details to logging on failure.