tfm.vision.augment.from_4d

Converts a 4D image back to ndims rank.