org.nd4j.linalg.jcublas.fft.ops

Class JCudaVectorFFT

java.lang.Object

org.nd4j.linalg.api.ops.BaseOp

org.nd4j.linalg.api.ops.BaseTransformOp

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

org.nd4j.linalg.jcublas.fft.ops.JCudaVectorFFT

All Implemented Interfaces:

Op, TransformOp

public class JCudaVectorFFT
extends VectorFFT

Uses jcuda's vector fft rather than the normal cpu based method

Author:

Adam Gibson

Field Summary

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

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
JCudaVectorFFT()
JCudaVectorFFT(INDArray x)
JCudaVectorFFT(INDArray x, INDArray y, INDArray z, int n, int fftLength)
JCudaVectorFFT(INDArray x, INDArray z, int fftLength)
JCudaVectorFFT(INDArray x, INDArray z, int n, int fftLength)
JCudaVectorFFT(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.VectorFFT

isPassThrough, name, op, op, op, op, op, op, op, op, opForDimension, opForDimension, setX, setY, 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, 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, x, y, z

Constructor Detail

JCudaVectorFFT

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

JCudaVectorFFT

public JCudaVectorFFT()

JCudaVectorFFT

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

JCudaVectorFFT

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

JCudaVectorFFT

public JCudaVectorFFT(INDArray x,
                      int fftLength)

JCudaVectorFFT

public JCudaVectorFFT(INDArray x)

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 VectorFFT