Function torch::fft::irfft2

• Defined in File fft.h

Function Documentation

inline Tensor torch::fft::irfft2(const Tensor &self, at::OptionalIntArrayRef s = std::nullopt, IntArrayRef dim = {-2, -1}, std::optional<std::string_view> norm = std::nullopt)

Computes the inverse of torch.fft.rfft2.

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

Example:

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