org.nd4j.linalg.fft

Class DefaultFFTInstance

java.lang.Object

org.nd4j.linalg.fft.BaseFFTInstance

org.nd4j.linalg.fft.DefaultFFTInstance

All Implemented Interfaces:

FFTInstance

public class DefaultFFTInstance
extends BaseFFTInstance

Default FFT instance that will work that is backend agnostic.

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
DefaultFFTInstance()

Method Summary

Modifier and Type	Method and Description
IComplexNDArray	fft(IComplexNDArray inputC, int numElements, int dimension) 1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector.
IComplexNDArray	fft(INDArray transform, int numElements, int dimension) FFT along a particular dimension
protected Op	getFftOp(INDArray arr, int n)
protected Op	getIfftOp(INDArray arr, int n)
IComplexNDArray	ifft(IComplexNDArray inputC) 1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector.
IComplexNDArray	ifft(IComplexNDArray inputC, int numElements, int dimension) 1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector.
IComplexNDArray	ifft(INDArray transform, int numElements) FFT along a particular dimension
IComplexNDArray	ifft(INDArray transform, int numElements, int dimension) IFFT along a particular dimension
protected IComplexNDArray	postProcess(IComplexNDArray result, int dimension)
protected IComplexNDArray	preProcess(IComplexNDArray result, IComplexNDArray transform, int n, int dimension)
IComplexNDArray	rawfft(IComplexNDArray transform, int n, int dimension) Underlying fft algorithm
IComplexNDArray	rawifft(IComplexNDArray transform, int n, int dimension) Underlying ifft impl

Methods inherited from class org.nd4j.linalg.fft.BaseFFTInstance

doFFt, doInnerFft, fft, fft, fft, fft, fftn, fftn, fftn, fftn, ifftn, ifftn, ifftn, ifftn, ifftn, irfft, irfft, irfftn, rawfftn, rawifft, rawifftn

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DefaultFFTInstance

public DefaultFFTInstance()

Method Detail

fft

public IComplexNDArray fft(INDArray transform,
                           int numElements,
                           int dimension)

FFT along a particular dimension

Parameters:

transform - the ndarray to op

numElements - the desired number of elements in each fft

Returns:

the ffted output

fft

public IComplexNDArray fft(IComplexNDArray inputC,
                           int numElements,
                           int dimension)

1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector. See matlab's fft2 for more information

Parameters:

inputC - the input to op

Returns:

the the discrete fourier op of the passed in input

ifft

public IComplexNDArray ifft(INDArray transform,
                            int numElements,
                            int dimension)

IFFT along a particular dimension

Parameters:

transform - the ndarray to op

numElements - the desired number of elements in each fft

dimension - the dimension to do fft along

Returns:

the iffted output

ifft

public IComplexNDArray ifft(IComplexNDArray inputC,
                            int numElements,
                            int dimension)

1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector. See matlab's fft2 for more information

Parameters:

inputC - the input to op

Returns:

the the discrete fourier op of the passed in input

ifft

public IComplexNDArray ifft(INDArray transform,
                            int numElements)

FFT along a particular dimension

Parameters:

transform - the ndarray to op

numElements - the desired number of elements in each fft

Returns:

the ffted output

ifft

public IComplexNDArray ifft(IComplexNDArray inputC)

1d discrete fourier op, note that this will throw an exception if the passed in input isn't a vector. See matlab's fft2 for more information

Parameters:

inputC - the input to op

Returns:

the the discrete fourier op of the passed in input

rawfft

public IComplexNDArray rawfft(IComplexNDArray transform,
                              int n,
                              int dimension)

Underlying fft algorithm

Parameters:

transform - the ndarray to op

n - the desired number of elements

dimension - the dimension to do fft along

Returns:

the transformed ndarray

rawifft

public IComplexNDArray rawifft(IComplexNDArray transform,
                               int n,
                               int dimension)

Description copied from interface: FFTInstance

Underlying ifft impl

Parameters:

transform - the ndarray to op

n - the desired number of elements

dimension - the dimension to do fft along

Returns:

postProcess

protected IComplexNDArray postProcess(IComplexNDArray result,
                                      int dimension)

preProcess

protected IComplexNDArray preProcess(IComplexNDArray result,
                                     IComplexNDArray transform,
                                     int n,
                                     int dimension)

getIfftOp

protected Op getIfftOp(INDArray arr,
                       int n)

getFftOp

protected Op getFftOp(INDArray arr,
                      int n)