Table Of Contents
Table Of Contents

fft

mxnet.ndarray.contrib.fft(data=None, compute_size=_Null, out=None, name=None, **kwargs)

Apply 1D FFT to input”

Note

fft is only available on GPU.

Currently accept 2 input data shapes: (N, d) or (N1, N2, N3, d), data can only be real numbers. The output data has shape: (N, 2*d) or (N1, N2, N3, 2*d). The format is: [real0, imag0, real1, imag1, …].

Example:

data = np.random.normal(0,1,(3,4))
out = mx.contrib.ndarray.fft(data = mx.nd.array(data,ctx = mx.gpu(0)))

Defined in src/operator/contrib/fft.cc:L56

Parameters:
  • data (NDArray) – Input data to the FFTOp.
  • compute_size (int, optional, default='128') – Maximum size of sub-batch to be forwarded at one time
  • out (NDArray, optional) – The output NDArray to hold the result.
Returns:

out – The output of this function.

Return type:

NDArray or list of NDArrays