Function torch::fft::fftn

• Defined in File fft.h

Function Documentation

inline Tensor torch::fft::fftn(const Tensor &self, at::OptionalIntArrayRef s = c10::nullopt, at::OptionalIntArrayRef dim = c10::nullopt, c10::optional<c10::string_view> norm = c10::nullopt)

Computes the N dimensional fast Fourier transform over given dimensions.

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

Example:

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