Closed andrevitorelli closed 3 years ago
Adding axes=[1,2] to the fftshifts so that we don't shift in the batch direction.
Solves issue #22
Adding axes=[1,2] to the fftshifts so that we don't shift in the batch direction.