def fft2(a, b=None, axes=(0, 1), threads=1, **kw): if b is None: return numpy.fft.fft2(a, axes=axes) else: b[:] = numpy.fft.fft2(a, axes=axes) return b