Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

Reference guides

Copy

View Markdown Download .md

tensorplay.fft

Fast Fourier Transforms

`tensorplay.fft.fft`	Computes the one-dimensional discrete Fourier transform.
`tensorplay.fft.ifft`	Computes the one-dimensional inverse discrete Fourier transform.
`tensorplay.fft.fft2`	Computes the two-dimensional discrete Fourier transform.
`tensorplay.fft.ifft2`	Computes the two-dimensional inverse discrete Fourier transform.
`tensorplay.fft.fftn`	Computes the N-dimensional discrete Fourier transform over `dim`.
`tensorplay.fft.ifftn`	Computes the N-dimensional inverse discrete Fourier transform.
`tensorplay.fft.rfft`	Computes the one-dimensional FFT of real input, one-sided output.
`tensorplay.fft.irfft`	Computes the inverse of `rfft()`; `n` is the output length.
`tensorplay.fft.rfft2`	Two-dimensional FFT of real input: `fft()` on leading dims, then `rfft()` on the final transformed dimension.
`tensorplay.fft.irfft2`	Inverse of `rfft2()`: `irfft()` on the final dimension first (`s[-1]` is the real output size), then `ifft()` on the rest.
`tensorplay.fft.rfftn`	N-dimensional FFT of real input; one-sided along the last listed dim.
`tensorplay.fft.irfftn`	Inverse of `rfftn()`; `s[-1]` is the real output size.
`tensorplay.fft.hfft`	Computes the 1-D FFT of a Hermitian-symmetric spectrum; real output.
`tensorplay.fft.ihfft`	Computes the inverse of `hfft()`; one-sided complex output.
`tensorplay.fft.hfft2`	Two-dimensional inverse of a Hermitian-symmetric spectrum; real output.
`tensorplay.fft.ihfft2`	Two-dimensional counterpart of `ihfft()`.
`tensorplay.fft.hfftn`	N-dimensional FFT of a Hermitian-symmetric spectrum; real output.
`tensorplay.fft.ihfftn`	Inverse of `hfftn()`: `ihfft()` along the final transformed dimension, then `fft()` over the remaining dimensions.

Helper Functions

`tensorplay.fft.fftfreq`	DFT sample frequencies (cycles/unit): `[0, 1, ..., n/2-1, -n/2, ..., -1] / (n*d)`.
`tensorplay.fft.rfftfreq`	Sample frequencies for `rfft()`/one-sided transforms: `[0..n//2] / (n*d)`.
`tensorplay.fft.fftshift`	Re-orders an N-D FFT output so the zero-frequency term is centered.
`tensorplay.fft.ifftshift`	Inverse of `fftshift()`; shifts by `-(n // 2)` (odd-safe).

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.cuda

tensorplay.cuda.StreamContext

tensorplay.futures

tensorplay.futures.Future