tf.signal.fftnd

ND fast Fourier transform.

tf.signal.fftnd(
    input_tensor, fft_length=None, axes=None, norm=None, name=None
)

Computes the n-dimensional discrete Fourier transform over designated dimensions of input. The designated dimensions of input are assumed to be the result of FFTND.

If fft_length[i]shape(input)[i], the input is padded with zeros. If fft_length is not given, the default shape(input) is used.

Axes mean the dimensions to perform the transform on. Default is to perform on all axes.

input A Tensor. Must be one of the following types: complex64, complex128. A complex tensor.
fft_length A Tensor of type int32. An int32 tensor. The FFT length for each dimension.
axes A Tensor of type int32. An int32 tensor with a same shape as fft_length. Axes to perform the transform.
name A name for the operation (optional).

A Tensor. Has the same type as input.