def sens_reduce(x): x = T.ifft2(x) return T.complex_mul(x, T.complex_conj(sens_maps)).sum(dim=1, keepdim=True)
def sens_expand(x): return T.fft2(T.complex_mul(x, sens_maps))