Function torch::fft::fft

• Defined in File fft.h

Function Documentation

inline Tensor torch::fft::fft(const Tensor &self, std::optional<SymInt> n = std::nullopt, int64_t dim = -1, std::optional<c10::string_view> norm = std::nullopt)

Computes the 1 dimensional fast Fourier transform over a given dimension.

See https://pytorch.org/docs/main/fft.html#torch.fft.fft.

Example:

auto t = torch::randn(128, dtype=kComplexDouble);
torch::fft::fft(t);