ENH: accept 3D niimg in NiftiMasker.transform