I see your point, but normally I prefer functions to return an image of the exact same shape as the input. We could 1) use squeeze and restore the shape before returning, or 2) issue a warning if a dimension is 1. What do you think?