It needs to be cleaned up because all inputs are converted to 4-D arrays, and expect 3 spatial dimensions. If inputs are 2D, the third dimension is a singleton - but it's necessary.

Good point on "any iterable" though, I'll convert this to an `ndarray`.