FFT

class shampoo.FFT(shape, float_precision, complex_precision, threads=2)[source] [edit on github]

Bases: object

Convenience wrapper around pyfftw.builders.fft2.

Parameters:

shape : tuple

Shape of the arrays which you will take the Fourier transforms of.

float_precision : dtype

complex_precision : dtype

threads : int, optional

This FFT implementation uses multithreading, with two threads by default.

Methods Summary

fft2(array) 2D Fourier transform.
ifft2(array) Inverse 2D Fourier transform.

Methods Documentation

fft2(array)[source] [edit on github]

2D Fourier transform.

Parameters:

array : ndarray (real)

Input array

Returns:

ft_array : ndarray (complex)

Fourier transform of the input array

ifft2(array)[source] [edit on github]

Inverse 2D Fourier transform.

Parameters:

array : ndarray

Input array

Returns:

ift_array : ndarray

Inverse Fourier transform of input array