merge_samples can now merge along iterations or chains.