def sens_reduce(self, x: torch.Tensor, sens_maps: torch.Tensor) -> torch.Tensor: return fastmri.complex_mul(fastmri.ifft2c(x), fastmri.complex_conj(sens_maps)).sum( dim=1, keepdim=True)
def sens_reduce(x): x = fastmri.ifft2c(x) return fastmri.complex_mul(x, fastmri.complex_conj(sens_maps)).sum( dim=1, keepdim=True)
def sens_expand(self, x: torch.Tensor, sens_maps: torch.Tensor) -> torch.Tensor: return fastmri.fft2c(fastmri.complex_mul(x, sens_maps))
def sens_expand(x): return fastmri.fft2c(fastmri.complex_mul(x, sens_maps))