org.nd4j.linalg.jcublas.fft.ops

Class JCudaVectorIFFT

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseTransformOp

org.nd4j.linalg.api.ops.impl.transforms.VectorIFFT

org.nd4j.linalg.jcublas.fft.ops.JCudaVectorIFFT

All Implemented Interfaces:

Op, TransformOp

public class JCudaVectorIFFT
extends VectorIFFT

Uses jcuda's vector ifft rather than normal cpu computation

Author:

Adam Gibson

Field Summary

Fields inherited from class org.nd4j.linalg.api.ops.impl.transforms.VectorIFFT

executed, fftLength

Fields inherited from class org.nd4j.linalg.api.ops.BaseOp

extraArgs, n, numProcessed, passThrough, x, y, z

Constructor Summary

Constructors

Constructor and Description
JCudaVectorIFFT()
JCudaVectorIFFT(INDArray x, INDArray y, INDArray z, int n, int fftLength)
JCudaVectorIFFT(INDArray x, INDArray z, int fftLength)
JCudaVectorIFFT(INDArray x, INDArray z, int n, int fftLength)
JCudaVectorIFFT(INDArray x, int fftLength)

Method Summary

Modifier and Type	Method and Description
void	exec()
cufftHandle	getHandle() Get the handle for the current thread

Methods inherited from class org.nd4j.linalg.api.ops.impl.transforms.VectorIFFT

isPassThrough, name, op, op, op, op, op, op, op, op, opForDimension, opForDimension, setX, setZ

Methods inherited from class org.nd4j.linalg.api.ops.BaseTransformOp

derivative

Methods inherited from class org.nd4j.linalg.api.ops.BaseOp

ensureProperVectors, exec, extraArgs, init, n, numProcessed, setN, setY, toString, x, y, z

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.linalg.api.ops.Op

exec, extraArgs, init, n, numProcessed, setN, setY, x, y, z

Constructor Detail

JCudaVectorIFFT

public JCudaVectorIFFT(INDArray x,
                       INDArray z,
                       int fftLength)

JCudaVectorIFFT

public JCudaVectorIFFT(INDArray x,
                       INDArray z,
                       int n,
                       int fftLength)

JCudaVectorIFFT

public JCudaVectorIFFT(INDArray x,
                       INDArray y,
                       INDArray z,
                       int n,
                       int fftLength)

JCudaVectorIFFT

public JCudaVectorIFFT(INDArray x,
                       int fftLength)

JCudaVectorIFFT

public JCudaVectorIFFT()

Method Detail

getHandle

public cufftHandle getHandle()

Get the handle for the current thread

Returns:

the handle for the current thread

exec

public void exec()

Specified by:

exec in interface Op

Overrides:

exec in class VectorIFFT