org.nd4j.linalg.convolution

Class DefaultConvolutionInstance

java.lang.Object

org.nd4j.linalg.convolution.BaseConvolution

org.nd4j.linalg.convolution.DefaultConvolutionInstance

All Implemented Interfaces:

ConvolutionInstance

public class DefaultConvolutionInstance
extends BaseConvolution

Default convolution instance (FFT based)

Author:

Adam Gibson

Constructor Summary

Constructors

Constructor and Description
DefaultConvolutionInstance()

Method Summary

Modifier and Type	Method and Description
IComplexNDArray	convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type, int[] axes) ND Convolution
INDArray	convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes) ND Convolution

Methods inherited from class org.nd4j.linalg.convolution.BaseConvolution

conv2d, conv2d, convn, convn

Methods inherited from class java.lang.Object

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

Constructor Detail

DefaultConvolutionInstance

public DefaultConvolutionInstance()

Method Detail

convn

public IComplexNDArray convn(IComplexNDArray input,
                             IComplexNDArray kernel,
                             Convolution.Type type,
                             int[] axes)

ND Convolution

Parameters:

input - the input to op

kernel - the kernel to op with

type - the type of convolution

axes - the axes to do the convolution along

Returns:

the convolution of the given input and kernel

convn

public INDArray convn(INDArray input,
                      INDArray kernel,
                      Convolution.Type type,
                      int[] axes)

ND Convolution

Parameters:

input - the input to op

kernel - the kernel to op with

type - the type of convolution

axes - the axes to do the convolution along

Returns:

the convolution of the given input and kernel