particle sample output arrays have wrong shape
The output datasets of sample_from_particles_system
and sample_particles_system_position
should have the same shape as the corresponding datasets from the checkpoint file.
At the moment they have an extra dimension of size 1.